dvir-bria commited on
Commit
fac36c2
1 Parent(s): 3d994fb

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -82,6 +82,7 @@ class Model:
82
  return self.base_model_id
83
 
84
  def load_controlnet_weight(self, task_name: str) -> None:
 
85
  if task_name == self.task_name:
86
  return
87
  if self.pipe is not None and hasattr(self.pipe, "controlnet"):
 
82
  return self.base_model_id
83
 
84
  def load_controlnet_weight(self, task_name: str) -> None:
85
+ print('Entered load_controlnet_weight....')
86
  if task_name == self.task_name:
87
  return
88
  if self.pipe is not None and hasattr(self.pipe, "controlnet"):