1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

add PreTrainedTokenizerFast in loadable classes

This commit is contained in:
patil-suraj
2022-06-10 10:56:33 +02:00
parent 573fbdff16
commit a75846379a

View File

@@ -40,6 +40,7 @@ LOADABLE_CLASSES = {
},
"transformers": {
"PreTrainedTokenizer": ["save_pretrained", "from_pretrained"],
"PreTrainedTokenizerFast": ["save_pretrained", "from_pretrained"],
"PreTrainedModel": ["save_pretrained", "from_pretrained"],
},
}