AlekseyCalvin commited on
Commit
0945218
1 Parent(s): 9c77a26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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