+ generate

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
NVIDIA · Oct 17, 2024 · 4e81b39 · 4e81b39
1 parent f4a2f0a
commit 4e81b39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/llm/api.py b/nemo/collections/llm/api.py
@@ -432,7 +432,7 @@ def export_ckpt(
     return io.export_ckpt(path, target, output_path, overwrite, load_connector)
 
 
-@run.cli.entrypoint(name="generate", namespace="llm")
+@cli_entrypoint(name="generate", namespace="llm")
 def generate(
     path: Union[Path, str],
     prompts: list[str],