JunhaoZhuang commited on
Commit
5cf8e7c
·
verified ·
1 Parent(s): 07a911b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
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
- load_ckpt()
 
 
 
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):