Spaces:
Runtime error
Runtime error
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 |
|