RamAnanth1 commited on
Commit
484221b
1 Parent(s): 28425ed

Replace file paths with HF file paths

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -68,8 +68,8 @@ class Model:
68
  config = OmegaConf.load("config_sketch.yaml")
69
  config.model.params.cond_stage_config.params.device = device
70
 
71
- base_model_file = model_path+"sd-v1-4.ckpt"
72
- sketch_adapter_file = model_path+"t2iadapter_sketch_sd14v1.pth"
73
  pidinet_file = model_path+"table5_pidinet.pth"
74
 
75
  subprocess.run(shlex.split(f'wget {base_model_file} -O models/sd-v1-4.ckpt'))
 
68
  config = OmegaConf.load("config_sketch.yaml")
69
  config.model.params.cond_stage_config.params.device = device
70
 
71
+ base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/blob/main/sd-v1-4.ckpt"
72
+ sketch_adapter_file = "https://huggingface.co/TencentARC/T2I-Adapter/blob/main/models/t2iadapter_sketch_sd14v1.pth"
73
  pidinet_file = model_path+"table5_pidinet.pth"
74
 
75
  subprocess.run(shlex.split(f'wget {base_model_file} -O models/sd-v1-4.ckpt'))