added gradio (#133)

* added gradio * added gradio
hpcaitech · Jun 13, 2024 · df483a5 · df483a5
1 parent 6c34b68
commit df483a5
Show file tree

Hide file tree

Showing 3 changed files with 211 additions and 272 deletions.
diff --git a/configs/opensora-v1-2/inference/sample.py b/configs/opensora-v1-2/inference/sample.py
@@ -25,14 +25,14 @@
 
 model = dict(
     type="STDiT3-XL/2",
-    from_pretrained=None,
+    from_pretrained="/mnt/jfs/sora_checkpoints/042-STDiT3-XL-2/epoch0-global_step7200/ema.pt",
     qk_norm=True,
     enable_flash_attn=True,
     enable_layernorm_kernel=True,
 )
 vae = dict(
     type="OpenSoraVAE_V1_2",
-    from_pretrained="pretrained_models/vae-pipeline",
+    from_pretrained="/mnt/jfs/sora_checkpoints/vae-pipeline",
     micro_frame_size=17,
     micro_batch_size=4,
 )