fix the bug of 600M online ckpt path;

Signed-off-by: lawrence-cj <[email protected]>
NVlabs · Dec 4, 2024 · d794502 · d794502
1 parent 747918b
commit d794502
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/bash/test_inference.sh b/tests/bash/test_inference.sh
@@ -3,7 +3,7 @@ set -e
 
 python scripts/inference.py \
     --config=configs/sana_config/1024ms/Sana_600M_img1024.yaml \
-    --model_path=hf://Efficient-Large-Model/Sana_600M_1024px/checkpoints/Sana_600M_1024px.pth
+    --model_path=hf://Efficient-Large-Model/Sana_600M_1024px/checkpoints/Sana_600M_1024px_MultiLing.pth
 
 
 python scripts/inference.py \