Update src/pipeline.py
Browse files- 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.
|
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()
|