Spaces:
Runtime error
Runtime error
Update app_docker.py
Browse files- 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",
|
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
|