multimodalart HF staff commited on
Commit
ce8156e
·
verified ·
1 Parent(s): 8009c77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import os
2
  from huggingface_hub import snapshot_download
3
  import spaces
 
 
4
  os.makedirs("/home/user/app/ckpts", exist_ok=True)
5
  snapshot_download(repo_id="Tencent-Hunyuan/HunyuanDiT", local_dir="/home/user/app/ckpts")
6
 
 
1
  import os
2
  from huggingface_hub import snapshot_download
3
  import spaces
4
+ import torch
5
+ torch.jit.script = lambda f: f
6
  os.makedirs("/home/user/app/ckpts", exist_ok=True)
7
  snapshot_download(repo_id="Tencent-Hunyuan/HunyuanDiT", local_dir="/home/user/app/ckpts")
8