Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
|
@@ -17,7 +17,7 @@ refiner = DiffusionPipeline.from_pretrained(
|
|
| 17 |
use_safetensors=True,
|
| 18 |
variant="fp16"
|
| 19 |
)
|
| 20 |
-
refiner.to("
|
| 21 |
|
| 22 |
# Function to generate the image
|
| 23 |
def generate_image(prompt, n_steps=20, high_noise_frac=0.8):
|
|
|
|
| 17 |
use_safetensors=True,
|
| 18 |
variant="fp16"
|
| 19 |
)
|
| 20 |
+
refiner.to("cpu") # Use GPU if available, else change to "cpu"
|
| 21 |
|
| 22 |
# Function to generate the image
|
| 23 |
def generate_image(prompt, n_steps=20, high_noise_frac=0.8):
|