AlekseyCalvin commited on
Commit
d381177
1 Parent(s): de2e902

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/main/custom_pypeline.py", torch_dtype=dtype
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