Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ def run_dev_turbo(prompt):
|
|
| 45 |
def run_schnell(prompt):
|
| 46 |
print("schnell")
|
| 47 |
pipe_schnell.to("cuda")
|
| 48 |
-
image = pipe_schnell(prompt).images[0]
|
| 49 |
return image
|
| 50 |
|
| 51 |
def run_parallel_models(prompt):
|
|
|
|
| 45 |
def run_schnell(prompt):
|
| 46 |
print("schnell")
|
| 47 |
pipe_schnell.to("cuda")
|
| 48 |
+
image = pipe_schnell(prompt, num_inference_steps=4).images[0]
|
| 49 |
return image
|
| 50 |
|
| 51 |
def run_parallel_models(prompt):
|