test with simpler prompt

2026-05-31 06:28:45 +00:00
parent 152af7295a
commit 5f98855141
1 changed files with 1 additions and 1 deletions
--- 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

 # =====================================================================