diff --git a/single_shot_inference.py b/single_shot_inference.py index f141b898..552357bd 100644 --- a/single_shot_inference.py +++ b/single_shot_inference.py @@ -46,7 +46,7 @@ from pathlib import Path CHECKPOINT_DIR = "/root/nvidia-meeting/DeepSeek-V4-Pro-NVFP4" MAX_NEW_TOKENS = 10 -PROMPT = "Paris is the capital of" +PROMPT = "The capital of France is" NUM_GPUS = 8 # =====================================================================