snapshot_download

app.py

@@ -13,18 +13,10 @@ from hyvideo.config import parse_args
 from hyvideo.inference import HunyuanVideoSampler
 from hyvideo.constants import NEGATIVE_PROMPT
 
-from huggingface_hub import hf_hub_download
+from huggingface_hub import snapshot_download
 
 if torch.cuda.device_count() > 0:
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="LICENSE", local_dir="ckpts")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="Notice", local_dir="ckpts")
-    #hf_hub_download(repo_id="tencent/HunyuanVideo", filename="README.md", local_dir="ckpts")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="config.json", local_dir="ckpts")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states.pt", local_dir="ckpts/hunyuan-video-t2v-720p/transformers")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states_fp8.pt", local_dir="ckpts/hunyuan-video-t2v-720p/transformers")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states_fp8_map.pt", local_dir="ckpts/hunyuan-video-t2v-720p/transformers")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/config.json", local_dir="ckpts/hunyuan-video-t2v-720p/vae")
-    hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/pytorch_model.pt", local_dir="ckpts/hunyuan-video-t2v-720p/vae")
+    snapshot_download(repo_id="tencent/HunyuanVideo", repo_type="model", local_dir="ckpts", force_download=True)
 
 def initialize_model(model_path):
     print('initialize_model: ' + model_path)
@@ -138,14 +130,14 @@ def create_demo(model_path):
                 
                 with gr.Accordion("Advanced Options", open=False):
                     with gr.Column():
-                        seed = gr.Slider(label="Seed (-1 for random)", value=-1, minimum=-1, maximum=2**63 - 1)
+                        seed = gr.Slider(label="Seed (-1 for random)", value=-1, minimum=-1, maximum=2**63 - 1, step=1)
                         guidance_scale = gr.Slider(1.0, 20.0, value=1.0, step=0.5, label="Guidance Scale")
                         flow_shift = gr.Slider(0.0, 10.0, value=7.0, step=0.1, label="Flow Shift") 
                         embedded_guidance_scale = gr.Slider(1.0, 20.0, value=6.0, step=0.5, label="Embedded Guidance Scale")
 
                 generate_btn = gr.Button(value = "🚀 Generate Video", variant = "primary")
             
-            with gr.Column():
+            with gr.Row():
                 output = gr.Video(label = "Generated Video", autoplay = True)
 
                 gr.Markdown("""