Spaces:

Princemmm
/

dave-ai-api

Configuration error

Princemmm commited on Jan 25

Commit

426962e

verified ·

1 Parent(s): df38773

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,7 @@ MODEL_ID = "hakurei/waifu-diffusion-v1-4"
 def generate(prompt):
     pipe = StableDiffusionPipeline.from_pretrained(
         MODEL_ID,
-        torch_dtype=torch.float16,
-        revision="fp16"
     )
     pipe = pipe.to("cpu")

 def generate(prompt):
     pipe = StableDiffusionPipeline.from_pretrained(
         MODEL_ID,
+        torch_dtype=torch.float16
     )
     pipe = pipe.to("cpu")