iimmortall commited on
Commit
0ed0c7c
·
verified ·
1 Parent(s): f5e7a54

Update app_docker.py

Browse files
Files changed (1) hide show
  1. app_docker.py +1 -1
app_docker.py CHANGED
@@ -14,7 +14,7 @@ from torchvision.transforms import ToTensor, ToPILImage
14
  from huggingface_hub import hf_hub_download, snapshot_download
15
  model_name = "iimmortall/UltraFusion"
16
  auth_token = os.getenv("HF_AUTH_TOKEN")
17
- model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app", resume_download=True, force_download=True)
18
 
19
  from ultrafusion_utils import load_model, run_ultrafusion, check_input
20
  PYCUDA_FLAG = True
 
14
  from huggingface_hub import hf_hub_download, snapshot_download
15
  model_name = "iimmortall/UltraFusion"
16
  auth_token = os.getenv("HF_AUTH_TOKEN")
17
+ model_folder = snapshot_download(repo_id=model_name, token=auth_token, local_dir="/home/user/app", force_download=True)
18
 
19
  from ultrafusion_utils import load_model, run_ultrafusion, check_input
20
  PYCUDA_FLAG = True