reverting

src/utils/download_models.py

@@ -5,13 +5,8 @@ from huggingface_hub import hf_hub_download
 
 
 def prepare_base_model():
-    base_dir = os.path.dirname(os.path.abspath(__file__))
-    config_path = os.path.join(base_dir, 'pretrained_weights', 'stable-diffusion-v1-5')
-
-    if not os.path.exists(config_path):
-        raise FileNotFoundError(f"The stable-diffusion-v1-5 folder was not found at: {config_path}")
-
-    os.makedirs(config_path, exist_ok=True)
+    local_dir = "./pretrained_weights/stable-diffusion-v1-5"
+    os.makedirs(local_dir, exist_ok=True)
     for hub_file in ["unet/config.json", "unet/diffusion_pytorch_model.bin"]:
         path = Path(hub_file)
         saved_path = local_dir / path
@@ -26,13 +21,8 @@ def prepare_base_model():
 
 
 def prepare_image_encoder():
-    base_dir = os.path.dirname(os.path.abspath(__file__))
-    config_path = os.path.join(base_dir, 'pretrained_weights')
-
-    if not os.path.exists(config_path):
-        raise FileNotFoundError(f"The pretrained_weights folder was not found at: {config_path}")
-
-    os.makedirs(config_path, exist_ok=True)
+    local_dir = "./pretrained_weights"
+    os.makedirs(local_dir, exist_ok=True)
     for hub_file in ["image_encoder/config.json", "image_encoder/pytorch_model.bin"]:
         path = Path(hub_file)
         saved_path = local_dir / path