From a75846379afc0557352be7df76780c3ac4aa113e Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Fri, 10 Jun 2022 10:56:33 +0200 Subject: [PATCH] add PreTrainedTokenizerFast in loadable classes --- src/diffusers/pipeline_utils.py | 1 + 1 file changed, 1 insertion(+) 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"], }, }