Playground-v1.0

Running on Zero

Yjiggfghhjnjj commited on May 5

Commit

d8e7f41

•

1 Parent(s): 14a9763

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ playground_pipe = DiffusionPipeline.from_pretrained(
         add_watermarker=False,
         variant="fp16"
     )
-playground.to("cuda")
 @spaces.GPU
 def run_comparison(prompt: str,

         add_watermarker=False,
         variant="fp16"
     )
+playground_pipe.to("cuda")
 @spaces.GPU
 def run_comparison(prompt: str,