Update src/pipeline.py

src/pipeline.py

@@ -31,7 +31,7 @@ def load_pipeline() -> Pipeline:
     transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16, use_safetensors=False,)
     pipeline = FluxPipeline.from_pretrained(CHECKPOINT, revision=REVISION, local_files_only=True, text_encoder=text_encoder, text_encoder_2=text_encoder_2, transformer=transformer, vae=vae, torch_dtype=torch.bfloat16,).to("cuda")
     quantize_(pipeline.vae, int8_weight_only())
-    pipeline = apply_cache_on_pipe(pipeline, residual_diff_threshold=0.55)
+    pipeline = apply_cache_on_pipe(pipeline, residual_diff_threshold=0.345)
     pipeline("")
     return pipeline
 @torch.no_grad()