DarkIdol-flux

Running on Zero

aifeifei798 commited on Nov 30, 2024

Commit

a1a01e0

verified ·

1 Parent(s): 4e9e36c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from diffusers import DiffusionPipeline, AutoencoderTiny
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
-    taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
         device
 )

 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
+taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
         device
 )