Spaces:

Idor980
/

text-to-image-template

Runtime error

Idor980 commited on May 28

Commit

1d05284

•

1 Parent(s): ad362b2

update space

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ if torch.cuda.is_available():
     pipe = pipe.to(device)
 else:
     pipe = DiffusionPipeline.from_pretrained(
-        "stabilityai/sdxl-turbo", use_safetensors=True
     )
     pipe = pipe.to(device)

     pipe = pipe.to(device)
 else:
     pipe = DiffusionPipeline.from_pretrained(
+        "stabilityai/sdxl-turbo", torch_dtype=torch.float16, use_safetensors=True
     )
     pipe = pipe.to(device)