Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
0945218
1
Parent(s):
9c77a26
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ torch._inductor.config.epilogue_fusion = False
|
|
35 |
torch._inductor.config.coordinate_descent_check_all_directions = True
|
36 |
|
37 |
dtype = torch.bfloat16
|
38 |
-
pipe = FluxWithCFGPipeline.from_pretrained("ostris/OpenFLUX.1", torch_dtype=dtype
|
39 |
).to("cuda")
|
40 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")
|
41 |
|
|
|
35 |
torch._inductor.config.coordinate_descent_check_all_directions = True
|
36 |
|
37 |
dtype = torch.bfloat16
|
38 |
+
pipe = FluxWithCFGPipeline.from_pretrained("ostris/OpenFLUX.1", torch_dtype=dtype, text_encoder_3=None, tokenizer_3=None
|
39 |
).to("cuda")
|
40 |
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")
|
41 |
|