kael558 commited on
Commit
5ad7e10
1 Parent(s): 959520f

changed repo name

Browse files
Files changed (1) hide show
  1. sd.py +3 -2
sd.py CHANGED
@@ -22,7 +22,7 @@ from omegaconf import OmegaConf
22
  from huggingface_hub import hf_hub_download
23
 
24
  # 1. Download stable diffusion repository and set path
25
- os.system("git clone https://github.com/kael558/stable-diffusion")
26
  os.system("git clone https://github.com/shariqfarooq123/AdaBins.git")
27
  os.system("git clone https://github.com/isl-org/MiDaS.git")
28
  os.system("git clone https://github.com/MSFTserver/pytorch3d-lite.git")
@@ -85,7 +85,8 @@ else:
85
 
86
  print('Model downloaded.')
87
  with open(os.path.join(models_path, model_checkpoint), 'wb') as model_file:
88
- model_file.write(ckpt_request.content)'''
 
89
 
90
  ckpt_path = hf_hub_download(repo_id="CompVis/stable-diffusion-v-1-4-original", filename="sd-v1-4.ckpt")
91
  ckpt_config_path = "./stable-diffusion/configs/stable-diffusion/v1-inference.yaml"
 
22
  from huggingface_hub import hf_hub_download
23
 
24
  # 1. Download stable diffusion repository and set path
25
+ os.system("git clone https://github.com/kael558/stable-diffusion-cpu")
26
  os.system("git clone https://github.com/shariqfarooq123/AdaBins.git")
27
  os.system("git clone https://github.com/isl-org/MiDaS.git")
28
  os.system("git clone https://github.com/MSFTserver/pytorch3d-lite.git")
 
85
 
86
  print('Model downloaded.')
87
  with open(os.path.join(models_path, model_checkpoint), 'wb') as model_file:
88
+ model_file.write(ckpt_request.content)
89
+ '''
90
 
91
  ckpt_path = hf_hub_download(repo_id="CompVis/stable-diffusion-v-1-4-original", filename="sd-v1-4.ckpt")
92
  ckpt_config_path = "./stable-diffusion/configs/stable-diffusion/v1-inference.yaml"