caohy666 commited on
Commit
d643b3f
·
1 Parent(s): 5fd1b45

<fix> fix a bug.

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,9 +45,6 @@ there's no need to manually input edge maps, depth maps, or other condition imag
45
  The corresponding condition images will be automatically extracted.
46
  """
47
 
48
- global pipe, current_task
49
- pipe = None
50
- current_task = None
51
 
52
  def init_basemodel():
53
  global transformer, scheduler, vae, text_encoder, text_encoder_2, tokenizer, tokenizer_2, image_processor
@@ -321,5 +318,8 @@ def create_app():
321
 
322
 
323
  if __name__ == "__main__":
 
 
 
324
  init_basemodel()
325
  create_app().launch(debug=True, ssr_mode=False)
 
45
  The corresponding condition images will be automatically extracted.
46
  """
47
 
 
 
 
48
 
49
  def init_basemodel():
50
  global transformer, scheduler, vae, text_encoder, text_encoder_2, tokenizer, tokenizer_2, image_processor
 
318
 
319
 
320
  if __name__ == "__main__":
321
+ global pipe, current_task
322
+ pipe = None
323
+ current_task = None
324
  init_basemodel()
325
  create_app().launch(debug=True, ssr_mode=False)