Skip to content

Commit 961b9b2

Browse files
authored
[docs] fix torchao typo. (#12883)
fix torchao typo.
1 parent 8f30bff commit 961b9b2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docs/source/en/quantization/torchao.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ pipeline_quant_config = PipelineQuantizationConfig(
3333
)
3434
pipeline = DiffusionPipeline.from_pretrained(
3535
"black-forest-labs/FLUX.1-dev",
36-
quantzation_config=pipeline_quant_config,
36+
quantization_config=pipeline_quant_config,
3737
torch_dtype=torch.bfloat16,
3838
device_map="cuda"
3939
)
@@ -50,7 +50,7 @@ pipeline_quant_config = PipelineQuantizationConfig(
5050
)
5151
pipeline = DiffusionPipeline.from_pretrained(
5252
"black-forest-labs/FLUX.1-dev",
53-
quantzation_config=pipeline_quant_config,
53+
quantization_config=pipeline_quant_config,
5454
torch_dtype=torch.bfloat16,
5555
device_map="cuda"
5656
)
@@ -70,7 +70,7 @@ pipeline_quant_config = PipelineQuantizationConfig(
7070
)
7171
pipeline = DiffusionPipeline.from_pretrained(
7272
"black-forest-labs/FLUX.1-dev",
73-
quantzation_config=pipeline_quant_config,
73+
quantization_config=pipeline_quant_config,
7474
torch_dtype=torch.bfloat16,
7575
device_map="cuda"
7676
)

0 commit comments

Comments
 (0)