diff --git a/docs/source/en/api/pipelines/mochi.md b/docs/source/en/api/pipelines/mochi.md index a997007a1b285..23133f9db660c 100644 --- a/docs/source/en/api/pipelines/mochi.md +++ b/docs/source/en/api/pipelines/mochi.md @@ -40,7 +40,7 @@ from transformers import BitsAndBytesConfig as BitsAndBytesConfig, T5EncoderMode quant_config = BitsAndBytesConfig(load_in_8bit=True) text_encoder_8bit = T5EncoderModel.from_pretrained( "genmo/mochi-1-preview", - subfolder="text_encoder2", + subfolder="text_encoder", quantization_config=quant_config, torch_dtype=torch.float16, )