Ming Li commited on
Commit
68f63f3
1 Parent(s): 3da112e

add Decorator for each condition

Browse files
Files changed (1) hide show
  1. model.py +0 -2
model.py CHANGED
@@ -44,7 +44,6 @@ class Model:
44
  self.pipe = self.load_pipe(base_model_id, task_name)
45
  self.preprocessor = Preprocessor()
46
 
47
- @spaces.GPU()
48
  def load_pipe(self, base_model_id: str, task_name) -> DiffusionPipeline:
49
  if (
50
  base_model_id == self.base_model_id
@@ -81,7 +80,6 @@ class Model:
81
  self.pipe = self.load_pipe(self.base_model_id, self.task_name)
82
  return self.base_model_id
83
 
84
- @spaces.GPU()
85
  def load_controlnet_weight(self, task_name: str) -> None:
86
  if task_name == self.task_name:
87
  return
 
44
  self.pipe = self.load_pipe(base_model_id, task_name)
45
  self.preprocessor = Preprocessor()
46
 
 
47
  def load_pipe(self, base_model_id: str, task_name) -> DiffusionPipeline:
48
  if (
49
  base_model_id == self.base_model_id
 
80
  self.pipe = self.load_pipe(self.base_model_id, self.task_name)
81
  return self.base_model_id
82
 
 
83
  def load_controlnet_weight(self, task_name: str) -> None:
84
  if task_name == self.task_name:
85
  return