Hugging Face's logo

jokerbit
/
flux-qa-5

Model card Files
xet
Community
jokerbit commited on
Commit
5fc050c
·
verified ·
1 Parent(s): 1f060f6

Heavy FBC

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -54,7 +54,7 @@ def load_pipeline() -> Pipeline:
54
  pipeline.to(memory_format=torch.channels_last)
55
  # quantize_(pipeline.vae, int8_weight_only())
56
  pipeline.vae = my_partial_compile(pipeline.vae)
57
- apply_cache_on_pipe(pipeline, residual_diff_threshold=0.25)
58
  with torch.inference_mode():
59
  for _ in range(2):
60
  pipeline("cats running on a road with a dog chasing", num_inference_steps=4)
 
54
  pipeline.to(memory_format=torch.channels_last)
55
  # quantize_(pipeline.vae, int8_weight_only())
56
  pipeline.vae = my_partial_compile(pipeline.vae)
57
+ apply_cache_on_pipe(pipeline, residual_diff_threshold=0.65)
58
  with torch.inference_mode():
59
  for _ in range(2):
60
  pipeline("cats running on a road with a dog chasing", num_inference_steps=4)