slobers commited on
Commit
adab011
·
verified ·
1 Parent(s): 3aab0e2

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -23,7 +23,7 @@ def load_pipeline() -> Pipeline:
23
  vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-schnell", revision="741f7c3ce8b383c54771c7003378a50191e9efe9", subfolder="vae", torch_dtype=torch.bfloat16)
24
  pipeline = FluxPipeline.from_pretrained(ids, revision=Revision, transformer=transformer, vae=vae, local_files_only=True, torch_dtype=torch.bfloat16)
25
  pipeline.to("cuda")
26
- pipeline = apply_cache_on_pipe(pipeline, residual_diff_threshold=0.888)
27
  pipeline("")
28
  return pipeline
29
  @torch.no_grad()
 
23
  vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-schnell", revision="741f7c3ce8b383c54771c7003378a50191e9efe9", subfolder="vae", torch_dtype=torch.bfloat16)
24
  pipeline = FluxPipeline.from_pretrained(ids, revision=Revision, transformer=transformer, vae=vae, local_files_only=True, torch_dtype=torch.bfloat16)
25
  pipeline.to("cuda")
26
+ pipeline = apply_cache_on_pipe(pipeline, residual_diff_threshold=0.768)
27
  pipeline("")
28
  return pipeline
29
  @torch.no_grad()