tencent_HunyuanVideo

gradio_server.py

@@ -12,6 +12,18 @@ from hyvideo.config import parse_args
 from hyvideo.inference import HunyuanVideoSampler
 from hyvideo.constants import NEGATIVE_PROMPT
 
+from huggingface_hub import hf_hub_download
+
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="LICENSE", local_dir="tencent_HunyuanVideo")
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="Notice", local_dir="tencent_HunyuanVideo")
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="README.md", local_dir="tencent_HunyuanVideo")
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="config.json", local_dir="tencent_HunyuanVideo")
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states.pt", local_dir="tencent_HunyuanVideo/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="tencent_HunyuanVideo/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="tencent_HunyuanVideo/hunyuan-video-t2v-720p/transformers")
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/config.json", local_dir="tencent_HunyuanVideo/hunyuan-video-t2v-720p/vae")
+hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/pytorch_model.pt", local_dir="tencent_HunyuanVideo/hunyuan-video-t2v-720p/vae")
+
 def initialize_model(model_path):
     args = parse_args()
     models_root_path = Path(model_path)
@@ -143,5 +155,5 @@ def create_demo(model_path):
 
 if __name__ == "__main__":
     os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
-    demo = create_demo("tencent/HunyuanVideo")
+    demo = create_demo("tencent_HunyuanVideo")
     demo.queue(10).launch()