Hugging Face's logo

jokerbit
/
flux-qa-base

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

Run

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -45,7 +45,7 @@ def load_pipeline() -> Pipeline:
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, fullgraph=True, mode="max-autotune")
49
  pipeline.to("cuda")
50
  for _ in range(4):
51
  pipeline("cat", num_inference_steps=4)
 
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, mode="max-autotune")
49
  pipeline.to("cuda")
50
  for _ in range(4):
51
  pipeline("cat", num_inference_steps=4)