limingcv commited on
Commit
00ff483
1 Parent(s): b2d51ed

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 = "benjamin-paine/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 = "botp/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 = ""