sayakpaul HF Staff commited on
Commit
d8e3cfe
·
verified ·
1 Parent(s): 7d0a465

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -14,8 +14,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
  # Load the model pipeline
16
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/Flux.1-Dev", torch_dtype=dtype).to(device)
17
- pipe.transformer.fuse_qkv_projections()
18
- pipe.transformer.set_attn_processor(FlashFusedFluxAttnProcessor3_0())
19
 
20
  @spaces.GPU(duration=1200)
21
  def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken):
 
14
 
15
  # Load the model pipeline
16
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/Flux.1-Dev", torch_dtype=dtype).to(device)
17
+ pipe.transformer.set_attn_processor(FlashFluxAttnProcessor3_0())
 
18
 
19
  @spaces.GPU(duration=1200)
20
  def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken):