playground-v2

Runtime error

ddosxd commited on Feb 27, 2024

Commit

5560523

verified ·

1 Parent(s): a4ea7a1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ NUM_IMAGES_PER_PROMPT = 1
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
-        "playgroundai/playground-v2-1024px-aesthetic",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,

 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
+        "playgroundai/playground-v2.5-1024px-aesthetic",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,