Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -175,15 +175,13 @@ image_encoder = CLIPVisionModelWithProjection.from_pretrained(os.path.join(model
|
|
175 |
# model_sketch.eval()
|
176 |
|
177 |
|
|
|
|
|
|
|
|
|
178 |
|
179 |
@spaces.GPU
|
180 |
def load_ckpt():
|
181 |
-
global pipeline
|
182 |
-
global MultiResNetModel
|
183 |
-
global causal_dit
|
184 |
-
global controlnet
|
185 |
-
global cur_style
|
186 |
-
cur_style = 'line + shadow'
|
187 |
print('load init',pipline)
|
188 |
weight_dtype = torch.float16
|
189 |
|
@@ -294,7 +292,10 @@ def load_ckpt():
|
|
294 |
|
295 |
print('loaded pipeline')
|
296 |
|
297 |
-
|
|
|
|
|
|
|
298 |
|
299 |
@spaces.GPU
|
300 |
def change_ckpt(style):
|
|
|
175 |
# model_sketch.eval()
|
176 |
|
177 |
|
178 |
+
global pipeline
|
179 |
+
global MultiResNetModel
|
180 |
+
global cur_style
|
181 |
+
cur_style = 'line + shadow'
|
182 |
|
183 |
@spaces.GPU
|
184 |
def load_ckpt():
|
|
|
|
|
|
|
|
|
|
|
|
|
185 |
print('load init',pipline)
|
186 |
weight_dtype = torch.float16
|
187 |
|
|
|
292 |
|
293 |
print('loaded pipeline')
|
294 |
|
295 |
+
return pipeline, MultiResNetModel
|
296 |
+
|
297 |
+
|
298 |
+
pipeline, MultiResNetModel = load_ckpt()
|
299 |
|
300 |
@spaces.GPU
|
301 |
def change_ckpt(style):
|