jokerbit
/

flux-qa-5

Model card Files Files and versions Community

jokerbit commited on Feb 13

Commit

5fc050c

·

verified ·

1 Parent(s): 1f060f6

Heavy FBC

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -54,7 +54,7 @@ def load_pipeline() -> Pipeline:
     pipeline.to(memory_format=torch.channels_last)
     # quantize_(pipeline.vae, int8_weight_only())
     pipeline.vae = my_partial_compile(pipeline.vae)
-    apply_cache_on_pipe(pipeline, residual_diff_threshold=0.25)
     with torch.inference_mode():
         for _ in range(2):
             pipeline("cats running on a road with a dog chasing", num_inference_steps=4)

     pipeline.to(memory_format=torch.channels_last)
     # quantize_(pipeline.vae, int8_weight_only())
     pipeline.vae = my_partial_compile(pipeline.vae)
+    apply_cache_on_pipe(pipeline, residual_diff_threshold=0.65)
     with torch.inference_mode():
         for _ in range(2):
             pipeline("cats running on a road with a dog chasing", num_inference_steps=4)