arthur-qiu commited on
Commit
3c096a1
1 Parent(s): b178d6b
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import os
2
  import torch
3
  from PIL import Image
@@ -5,9 +8,6 @@ from PIL import Image
5
  from pipeline_freescale import StableDiffusionXLPipeline
6
  from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
7
 
8
- import gradio as gr
9
- import spaces
10
-
11
  @spaces.GPU(duration=120)
12
  def infer_gpu_part(pipe, generator, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale):
13
  pipe = pipe.to("cuda")
 
1
+ import gradio as gr
2
+ import spaces
3
+
4
  import os
5
  import torch
6
  from PIL import Image
 
8
  from pipeline_freescale import StableDiffusionXLPipeline
9
  from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
10
 
 
 
 
11
  @spaces.GPU(duration=120)
12
  def infer_gpu_part(pipe, generator, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale):
13
  pipe = pipe.to("cuda")