Spaces:
Sleeping
Sleeping
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,
|