Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -175,14 +175,14 @@ image_encoder = CLIPVisionModelWithProjection.from_pretrained(os.path.join(model
|
|
| 175 |
# model_sketch.eval()
|
| 176 |
|
| 177 |
|
| 178 |
-
pipeline = None
|
| 179 |
-
MultiResNetModel = None
|
| 180 |
-
causal_dit = None
|
| 181 |
-
controlnet = None
|
| 182 |
-
cur_style = 'line + shadow'
|
| 183 |
|
| 184 |
@spaces.GPU
|
| 185 |
def load_ckpt():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 186 |
print('load init',pipline)
|
| 187 |
weight_dtype = torch.float16
|
| 188 |
|
|
|
|
| 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 = 'line + shadow'
|
| 186 |
print('load init',pipline)
|
| 187 |
weight_dtype = torch.float16
|
| 188 |
|