Update nemo_inference.sh
Browse files- nemo_inference.sh +1 -1
nemo_inference.sh
CHANGED
|
@@ -8,7 +8,7 @@ python3 /opt/NeMo/examples/nlp/language_modeling/megatron_gpt_eval.py \
|
|
| 8 |
inference.tokens_to_generate=100 \
|
| 9 |
inference.top_k=50 \
|
| 10 |
inference.top_p=0.7 \
|
| 11 |
-
inference.temperature=0.4
|
| 12 |
inference.add_BOS=False \
|
| 13 |
gpt_model_file="model.nemo" \
|
| 14 |
prompts="['<user>Capital of India is<assistant>']"
|
|
|
|
| 8 |
inference.tokens_to_generate=100 \
|
| 9 |
inference.top_k=50 \
|
| 10 |
inference.top_p=0.7 \
|
| 11 |
+
inference.temperature=0.4 \
|
| 12 |
inference.add_BOS=False \
|
| 13 |
gpt_model_file="model.nemo" \
|
| 14 |
prompts="['<user>Capital of India is<assistant>']"
|