Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
484221b
1
Parent(s):
28425ed
Replace file paths with HF file paths
Browse files
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 =
|
72 |
-
sketch_adapter_file =
|
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'))
|