Use model ID and add logs

app.py

@@ -13,29 +13,32 @@ 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):
-    if torch.cuda.device_count() == 0:
-        return None
+    print('initialize_model: ' + model_path)
+    #if torch.cuda.device_count() == 0:
+        #return None
     
     args = parse_args()
     models_root_path = Path(model_path)
     if not models_root_path.exists():
         raise ValueError(f"`models_root` not exists: {models_root_path}")
     
+    print(f"`models_root` exists: {models_root_path}")
     hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained(models_root_path, args=args)
+    print('Model initialized: ' + model_path)
     return hunyuan_video_sampler
 
 @spaces.GPU(duration=120)
@@ -164,5 +167,5 @@ def create_demo(model_path):
 
 if __name__ == "__main__":
     os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
-    demo = create_demo("ckpts")
+    demo = create_demo("tencent/HunyuanVideo")
     demo.queue(10).launch()