Download again

app.py

@@ -13,23 +13,23 @@ 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 hf_hub_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")
+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")
+    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")
 
 def initialize_model(model_path):
     print('initialize_model: ' + model_path)
-    #if torch.cuda.device_count() == 0:
-        #return None
+    if torch.cuda.device_count() == 0:
+        return None
     
     args = parse_args()
     models_root_path = Path(model_path)
@@ -167,5 +167,5 @@ def create_demo(model_path):
 
 if __name__ == "__main__":
     os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
-    demo = create_demo("tencent/HunyuanVideo")
+    demo = create_demo("ckpts")
     demo.queue(10).launch()