Hugging Face's logo

Spaces:
briaai
/
BRIA-2.2-ControlNet-Canny
Running on Zero

App Files Community
dvir-bria commited on
Commit
48401f8
•
1 Parent(s): 0d063fc

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -66,6 +66,7 @@ class Model:
66
  gc.collect()
67
  self.base_model_id = base_model_id
68
  self.task_name = task_name
 
69
  return pipe
70
 
71
  def set_base_model(self, base_model_id: str) -> str:
 
66
  gc.collect()
67
  self.base_model_id = base_model_id
68
  self.task_name = task_name
69
+ print(f'Loaded {base_model_id}...')
70
  return pipe
71
 
72
  def set_base_model(self, base_model_id: str) -> str: