Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
d381177
1
Parent(s):
de2e902
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ torch.backends.cuda.matmul.allow_tf32 = True
|
|
31 |
|
32 |
dtype = torch.bfloat16
|
33 |
pipe = FluxWithCFGPipeline.from_pretrained(
|
34 |
-
"ostris/OpenFLUX.1", custom_pipeline="AlekseyCalvin/OpenFlux_Lorasoonr
|
35 |
).to("cuda")
|
36 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")
|
37 |
|
|
|
31 |
|
32 |
dtype = torch.bfloat16
|
33 |
pipe = FluxWithCFGPipeline.from_pretrained(
|
34 |
+
"ostris/OpenFLUX.1", custom_pipeline="AlekseyCalvin/OpenFlux_Lorasoonr", torch_dtype=dtype
|
35 |
).to("cuda")
|
36 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")
|
37 |
|