before merge from main

NVIDIA · May 17, 2024 · 61faeb4 · 61faeb4
1 parent 356c79e
commit 61faeb4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/nlp/rag/rag_eval.py b/examples/nlp/rag/rag_eval.py
@@ -8,7 +8,7 @@
 from nemo.core.config import hydra_runner
 
 
-@hydra_runner(config_path="conf", config_name="rasg")
+@hydra_runner(config_path="conf", config_name="rag")
 def main(cfg) -> None:
 
     # load LLM

diff --git a/nemo/collections/nlp/models/rag/custom_llm.py b/nemo/collections/nlp/models/rag/custom_llm.py
@@ -24,7 +24,7 @@ class NeMoLLM(CustomLLM):
     dummy_response: str = "My response"
 
     length_params: LengthParam = {
-        "max_length": 30,
+        "max_length": 500,
         "min_length": 0,
     }