diff --git a/src/diffusers/pipeline_utils.py b/src/diffusers/pipeline_utils.py index aa67ccb154..d41832c32d 100644 --- a/src/diffusers/pipeline_utils.py +++ b/src/diffusers/pipeline_utils.py @@ -40,6 +40,7 @@ LOADABLE_CLASSES = { }, "transformers": { "PreTrainedTokenizer": ["save_pretrained", "from_pretrained"], + "PreTrainedTokenizerFast": ["save_pretrained", "from_pretrained"], "PreTrainedModel": ["save_pretrained", "from_pretrained"], }, }