Hugging Face's logo

jokerbit
/
flux-qa-base

Model card Files Community
jokerbit commited on
Commit
e4bd22a
·
verified ·
1 Parent(s): d3059b8

Fuse

Browse files
Files changed (1) hide show
  1. src/pipeline.py +2 -2
src/pipeline.py CHANGED
@@ -44,8 +44,8 @@ def load_pipeline() -> Pipeline:
44
  )
45
 
46
  pipeline.transformer.to(memory_format=torch.channels_last)
47
- # pipeline.vae.to(memory_format=torch.channels_last)
48
- pipeline.vae = torch.compile(pipeline.vae)
49
  pipeline.to("cuda")
50
  for _ in range(4):
51
  pipeline("cat", num_inference_steps=4)
 
44
  )
45
 
46
  pipeline.transformer.to(memory_format=torch.channels_last)
47
+ pipeline.text_encoder.fuse_qkv_projections()
48
+ # pipeline.vae = torch.compile(pipeline.vae)
49
  pipeline.to("cuda")
50
  for _ in range(4):
51
  pipeline("cat", num_inference_steps=4)