limingcv commited on
Commit
16ea5d7
1 Parent(s): e983349

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -37,7 +37,7 @@ def download_all_controlnet_weights() -> None:
37
 
38
 
39
  class Model:
40
- def __init__(self, base_model_id: str = "runwayml/stable-diffusion-v1-5", task_name: str = "Canny"):
41
  self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
42
  self.base_model_id = ""
43
  self.task_name = ""
 
37
 
38
 
39
  class Model:
40
+ def __init__(self, base_model_id: str = "ckpt/sd15", task_name: str = "Canny"):
41
  self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
42
  self.base_model_id = ""
43
  self.task_name = ""