Spaces:
Running
Running
Ming Li
commited on
Commit
•
68f63f3
1
Parent(s):
3da112e
add Decorator for each condition
Browse files
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
|