Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
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 |
|