KingNish commited on
Commit
3417772
1 Parent(s): 951bbb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -24,7 +24,6 @@ pipe = FluxWithCFGPipeline.from_pretrained(
24
  )
25
  pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
26
  pipe.to("cuda")
27
- pipe.unet.set_attn_processor(AttnProcessor2_0())
28
 
29
  torch.cuda.empty_cache()
30
 
 
24
  )
25
  pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
26
  pipe.to("cuda")
 
27
 
28
  torch.cuda.empty_cache()
29