changed ckpt to download from hf

.gitignore

@@ -2,7 +2,6 @@ venv/
 AdaBins/
 k-diffusion/
 MiDaS/
-models/*
 pytorch3d-lite/
 stable-diffusion/
 .idea

app.py

@@ -12,6 +12,7 @@ from types import SimpleNamespace
 import sd
 
 import warnings
+
 warnings.filterwarnings("ignore")
 
 #3. Set args

sd.py

@@ -18,8 +18,8 @@ from PIL import Image
 import os
 import requests
 
-#from huggingface_hub import snapshot_download
 from omegaconf import OmegaConf
+from huggingface_hub import hf_hub_download
 
 # 1. Download stable diffusion repository and set path
 os.system("git clone https://github.com/deforum/stable-diffusion")
@@ -72,12 +72,11 @@ def load_model_from_config(config, ckpt, verbose=False, device='cuda', half_prec
 
 print('Model loading...')
 
+'''
 models_path = "models"
 model_checkpoint =  "sd-v1-4.ckpt"
 ckpt_path = os.path.join(models_path, model_checkpoint)
-#ckpt_path = snapshot_download(repo_id="CompVis/stable-diffusion-v-1-4-original", filename="sd-v1-4.ckpt")
-
-ckpt_config_path = "./stable-diffusion/configs/stable-diffusion/v1-inference.yaml"
+ckpt_path = snapshot_download(repo_id="CompVis/stable-diffusion-v-1-4-original", filename="sd-v1-4.ckpt")
 
 if os.path.exists(ckpt_path):
     print(f"{ckpt_path} exists")
@@ -88,7 +87,10 @@ else:
 
     print('Model downloaded.')
     with open(os.path.join(models_path, model_checkpoint), 'wb') as model_file:
-        model_file.write(ckpt_request.content)
+        model_file.write(ckpt_request.content)'''
+
+ckpt_path = hf_hub_download(repo_id="CompVis/stable-diffusion-v-1-4-original", filename="sd-v1-4.ckpt")
+ckpt_config_path = "./stable-diffusion/configs/stable-diffusion/v1-inference.yaml"
 
 local_config = OmegaConf.load(f"{ckpt_config_path}")
 model = load_model_from_config(local_config, f"{ckpt_path}")