Hugging Face's logo

jokerbit
/
flux-qa-base

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

tiling

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -40,7 +40,7 @@ def load_pipeline() -> Pipeline:
40
  )
41
 
42
  pipeline.transformer.to(memory_format=torch.channels_last)
43
- pipeline.enable_vae_slicing()
44
  pipeline.to("cuda")
45
  for _ in range(4):
46
  pipeline("cat", num_inference_steps=4)
 
40
  )
41
 
42
  pipeline.transformer.to(memory_format=torch.channels_last)
43
+ pipeline.enable_vae_tiling()
44
  pipeline.to("cuda")
45
  for _ in range(4):
46
  pipeline("cat", num_inference_steps=4)