more directory updates

src/utils/download_models.py

@@ -5,8 +5,13 @@ from huggingface_hub import hf_hub_download
 
 
 def prepare_base_model():
-    local_dir = "./pretrained_weights/stable-diffusion-v1-5"
-    os.makedirs(local_dir, exist_ok=True)
+    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)
     for hub_file in ["unet/config.json", "unet/diffusion_pytorch_model.bin"]:
         path = Path(hub_file)
         saved_path = local_dir / path
@@ -21,8 +26,13 @@ def prepare_base_model():
 
 
 def prepare_image_encoder():
-    local_dir = "./pretrained_weights"
-    os.makedirs(local_dir, exist_ok=True)
+    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)
     for hub_file in ["image_encoder/config.json", "image_encoder/pytorch_model.bin"]:
         path = Path(hub_file)
         saved_path = local_dir / path