Unverified Commit df355ea2 authored by Omar Awile's avatar Omar Awile Committed by GitHub
Browse files

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.
parent ae019da9
...@@ -367,7 +367,7 @@ transformer_8bit = FluxTransformer2DModel.from_pretrained( ...@@ -367,7 +367,7 @@ transformer_8bit = FluxTransformer2DModel.from_pretrained(
pipeline = FluxPipeline.from_pretrained( pipeline = FluxPipeline.from_pretrained(
"black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-dev",
text_encoder=text_encoder_8bit, text_encoder_2=text_encoder_8bit,
transformer=transformer_8bit, transformer=transformer_8bit,
torch_dtype=torch.float16, torch_dtype=torch.float16,
device_map="balanced", device_map="balanced",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment