Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -24,6 +24,7 @@ pipe = fetch_model()
|
|
24 |
pipe.to('cuda')
|
25 |
|
26 |
|
|
|
27 |
def inference(user_state, condition_image, settings, prompt, neg_prompt, inference_steps=8, num_images=2,
|
28 |
guidance_scale=2.0,
|
29 |
guidance_rescale=0.0,
|
|
|
24 |
pipe.to('cuda')
|
25 |
|
26 |
|
27 |
+
@spaces.GPU
|
28 |
def inference(user_state, condition_image, settings, prompt, neg_prompt, inference_steps=8, num_images=2,
|
29 |
guidance_scale=2.0,
|
30 |
guidance_rescale=0.0,
|