nvidia
/

nemo-megatron-gpt-1.3B

Text2Text Generation

Model card Files Files and versions Community

okuchaiev commited on Sep 20, 2022

Commit

c46c6ff

•

1 Parent(s): 245571b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ Alternatively, you can use NeMo Megatron training docker container with all depe
 git clone https://github.com/NVIDIA/NeMo.git
 cd NeMo/examples/nlp/language_modeling
 git checkout v1.11.0
-python megatron_gpt_eval.py gpt_model_file=nemo_gpt5B_fp16.nemo server=True tensor_model_parallel_size=1 trainer.devices=1
 ```
 ### Step 3: Send prompts to you model!

 git clone https://github.com/NVIDIA/NeMo.git
 cd NeMo/examples/nlp/language_modeling
 git checkout v1.11.0
+python megatron_gpt_eval.py gpt_model_file=nemo_gpt1.3B_fp16.nemo server=True tensor_model_parallel_size=1 trainer.devices=1
 ```
 ### Step 3: Send prompts to you model!