Skip to content

Commit 6cad1d5

Browse files
committed
update
1 parent bb7fb66 commit 6cad1d5

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

docs/source/en/quantization/quanto.md

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,12 @@ from diffusers import FluxTransformer2DModel, QuantoConfig
5656

5757
model_id = "black-forest-labs/FLUX.1-dev"
5858
quantization_config = QuantoConfig(weights_dtype="float8", modules_to_not_convert=["proj_out"])
59-
transformer = FluxTransformer2DModel.from_pretrained(model_id, quantization_config=quantization_config, torch_dtype=torch.bfloat16)
59+
transformer = FluxTransformer2DModel.from_pretrained(
60+
model_id,
61+
subfolder="transformer",
62+
quantization_config=quantization_config,
63+
torch_dtype=torch.bfloat16,
64+
)
6065
```
6166

6267
## Using `from_single_file` with the Quanto Backend

0 commit comments

Comments
 (0)