File downloads

app.py

@@ -3,6 +3,15 @@ import types
 torch.cuda.get_device_capability = lambda *args, **kwargs: (8, 6)
 torch.cuda.get_device_properties = lambda *args, **kwargs: types.SimpleNamespace(name='NVIDIA A10G', major=8, minor=6, total_memory=23836033024, multi_processor_count=80)
 
+import huggingface_hub
+huggingface_hub.snapshot_download(repo_id='camenduru/PASD', local_dir='PASD/runs', allow_patterns=[
+    'pasd/**',
+    'pasd_light/**',
+    'pasd_light_rrdb/**',
+    'pasd_rrdb/**',
+])
+huggingface_hub.hf_hub_download(repo_id='camenduru/PASD', local_dir='PASD/checkpoints/personalized_models', filename='majicmixRealistic_v6.safetensors')
+
 import sys; sys.path.append('./PASD')
 import spaces
 import os
@@ -38,7 +47,7 @@ else:
     from models.pasd.unet_2d_condition import UNet2DConditionModel
     from models.pasd.controlnet import ControlNetModel
 
-pretrained_model_path = "checkpoints/stable-diffusion-v1-5"
+pretrained_model_path = "runwayml/stable-diffusion-v1-5"
 ckpt_path = "runs/pasd/checkpoint-100000"
 #dreambooth_lora_path = "checkpoints/personalized_models/toonyou_beta3.safetensors"
 dreambooth_lora_path = "checkpoints/personalized_models/majicmixRealistic_v6.safetensors"