playground-v2.5

Running on Zero

Yjiggfghhjnjj commited on May 5

Commit

f9bf6eb

•

1 Parent(s): cbd7579

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     pipe = DiffusionPipeline.from_pretrained(
         "playgroundai/playground-v2.5-1024px-aesthetic",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,
         variant="fp16"