Spaces:

OzzyGT
/

diffusers-image-fill

Running on Zero

OzzyGT HF Staff commited on Aug 26

Commit

80401cc

1 Parent(s): 432e607

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ prompt = "high quality"
     negative_prompt_embeds,
     pooled_prompt_embeds,
     negative_pooled_prompt_embeds,
-) = pipe.encode_prompt(prompt, "cuda")
 @spaces.GPU(duration=16)

     negative_prompt_embeds,
     pooled_prompt_embeds,
     negative_pooled_prompt_embeds,
+) = pipe.encode_prompt(prompt, device="cuda")
 @spaces.GPU(duration=16)