Hugging Face's logo

Spaces:
3bdull4h
/
HunyuanVideo
Running

App Files Community
Fabrice-TIERCELIN commited on
Commit
48af733
·
verified ·
1 Parent(s): 8029cd1

tencent_HunyuanVideo -> ckpts

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -16,15 +16,15 @@ from hyvideo.constants import NEGATIVE_PROMPT
16
  from huggingface_hub import hf_hub_download
17
 
18
  if torch.cuda.device_count() > 0:
19
- hf_hub_download(repo_id="tencent/HunyuanVideo", filename="LICENSE", local_dir="tencent_HunyuanVideo")
20
- hf_hub_download(repo_id="tencent/HunyuanVideo", filename="Notice", local_dir="tencent_HunyuanVideo")
21
- hf_hub_download(repo_id="tencent/HunyuanVideo", filename="README.md", local_dir="tencent_HunyuanVideo")
22
- hf_hub_download(repo_id="tencent/HunyuanVideo", filename="config.json", local_dir="tencent_HunyuanVideo")
23
- 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")
24
- 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")
25
- 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")
26
- hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/config.json", local_dir="tencent_HunyuanVideo/hunyuan-video-t2v-720p/vae")
27
- 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")
28
 
29
  def initialize_model(model_path):
30
  if torch.cuda.device_count() == 0:
@@ -164,5 +164,5 @@ def create_demo(model_path):
164
 
165
  if __name__ == "__main__":
166
  os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
167
- demo = create_demo("tencent_HunyuanVideo")
168
  demo.queue(10).launch()
 
16
  from huggingface_hub import hf_hub_download
17
 
18
  if torch.cuda.device_count() > 0:
19
+ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="LICENSE", local_dir="ckpts")
20
+ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="Notice", local_dir="ckpts")
21
+ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="README.md", local_dir="ckpts")
22
+ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="config.json", local_dir="ckpts")
23
+ 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")
24
+ 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")
25
+ 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")
26
+ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/config.json", local_dir="ckpts/hunyuan-video-t2v-720p/vae")
27
+ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/pytorch_model.pt", local_dir="ckpts/hunyuan-video-t2v-720p/vae")
28
 
29
  def initialize_model(model_path):
30
  if torch.cuda.device_count() == 0:
 
164
 
165
  if __name__ == "__main__":
166
  os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
167
+ demo = create_demo("ckpts")
168
  demo.queue(10).launch()