diff --git a/single_shot_inference.py b/single_shot_inference.py index fcc1be3b..650e11be 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 = "The capital of France is" +PROMPT = "Paris is the capital of" NUM_GPUS = 8 # =====================================================================