From df355ea2c657ca52d31b9c8e235436ce5f8da7bd Mon Sep 17 00:00:00 2001 From: Omar Awile Date: Mon, 13 Jan 2025 20:56:32 +0100 Subject: [PATCH] Fix documentation for FluxPipeline (#10563) Fix argument name in 8bit quantized example Found a tiny mistake in the documentation where the text encoder model was passed to the wrong argument in the FluxPipeline.from_pretrained function. --- docs/source/en/api/pipelines/flux.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/api/pipelines/flux.md b/docs/source/en/api/pipelines/flux.md index fd2c07e59f..f6e524af88 100644 --- a/docs/source/en/api/pipelines/flux.md +++ b/docs/source/en/api/pipelines/flux.md @@ -367,7 +367,7 @@ transformer_8bit = FluxTransformer2DModel.from_pretrained( pipeline = FluxPipeline.from_pretrained( "black-forest-labs/FLUX.1-dev", - text_encoder=text_encoder_8bit, + text_encoder_2=text_encoder_8bit, transformer=transformer_8bit, torch_dtype=torch.float16, device_map="balanced",