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):
|