Hugging Face's logo

jokerbit
/
flux-qa-1

Model card Files Community
jokerbit commited on
Commit
40b32d6
·
verified ·
1 Parent(s): 72968f9

Upload src/pipeline.py with huggingface_hub

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