Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ import spaces
|
|
| 11 |
import torch
|
| 12 |
from diffusers import DiffusionPipeline
|
| 13 |
|
| 14 |
-
DESCRIPTION = """# Playground v2"""
|
| 15 |
if not torch.cuda.is_available():
|
| 16 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
|
| 17 |
|
|
@@ -27,7 +27,7 @@ NUM_IMAGES_PER_PROMPT = 1
|
|
| 27 |
|
| 28 |
if torch.cuda.is_available():
|
| 29 |
pipe = DiffusionPipeline.from_pretrained(
|
| 30 |
-
"playgroundai/playground-v2-1024px-aesthetic",
|
| 31 |
torch_dtype=torch.float16,
|
| 32 |
use_safetensors=True,
|
| 33 |
add_watermarker=False,
|
|
|
|
| 11 |
import torch
|
| 12 |
from diffusers import DiffusionPipeline
|
| 13 |
|
| 14 |
+
DESCRIPTION = """# Playground v2.5"""
|
| 15 |
if not torch.cuda.is_available():
|
| 16 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
|
| 17 |
|
|
|
|
| 27 |
|
| 28 |
if torch.cuda.is_available():
|
| 29 |
pipe = DiffusionPipeline.from_pretrained(
|
| 30 |
+
"playgroundai/playground-v2.5-1024px-aesthetic",
|
| 31 |
torch_dtype=torch.float16,
|
| 32 |
use_safetensors=True,
|
| 33 |
add_watermarker=False,
|