[gradio] updated ckpt path (#141)

hpcaitech · Jun 16, 2024 · 7f7e989 · 7f7e989
1 parent b7e1c3f
commit 7f7e989
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gradio/app.py b/gradio/app.py
@@ -26,8 +26,8 @@
 }
 HF_STDIT_MAP = {
     "v1.2-stage3": {
-        "ema": "/mnt/jfs-hdd/sora/checkpoints/outputs/042-STDiT3-XL-2/epoch1-global_step18800/ema.pt",
-        "model": "/mnt/jfs-hdd/sora/checkpoints/outputs/042-STDiT3-XL-2/epoch1-global_step18800/model"
+        "ema": "/mnt/jfs-hdd/sora/checkpoints/outputs/042-STDiT3-XL-2/epoch1-global_step16200/ema.pt",
+        "model": "/mnt/jfs-hdd/sora/checkpoints/outputs/042-STDiT3-XL-2/epoch1-global_step16200/model"
         }
 }