Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -92,8 +92,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
92 |
|
93 |
with gr.Column(scale=4):
|
94 |
output = gr.Image(label="Your Generated Image")
|
95 |
-
|
96 |
-
@spaces.GPU
|
97 |
def process_image(height, width, steps, scales, prompt, seed, preset_prompt):
|
98 |
global pipe
|
99 |
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
|
|
|
92 |
|
93 |
with gr.Column(scale=4):
|
94 |
output = gr.Image(label="Your Generated Image")
|
95 |
+
|
96 |
+
@spaces.GPU
|
97 |
def process_image(height, width, steps, scales, prompt, seed, preset_prompt):
|
98 |
global pipe
|
99 |
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
|