Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,11 +10,10 @@ from PIL import Image, ImageDraw, ImageFont
|
|
| 10 |
# Load Stable Diffusion with Euler Scheduler
|
| 11 |
# =====================================================
|
| 12 |
model_id = "stabilityai/stable-diffusion-2-1-base"
|
| 13 |
-
|
| 14 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
| 15 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 16 |
model_id, scheduler=scheduler, torch_dtype=torch.float16
|
| 17 |
-
).to("cuda") #
|
| 18 |
|
| 19 |
# =====================================================
|
| 20 |
# Comic Styles & Descriptions
|
|
|
|
| 10 |
# Load Stable Diffusion with Euler Scheduler
|
| 11 |
# =====================================================
|
| 12 |
model_id = "stabilityai/stable-diffusion-2-1-base"
|
|
|
|
| 13 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
| 14 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 15 |
model_id, scheduler=scheduler, torch_dtype=torch.float16
|
| 16 |
+
).to("cuda") # Change to "cpu" if GPU unavailable
|
| 17 |
|
| 18 |
# =====================================================
|
| 19 |
# Comic Styles & Descriptions
|