diff --git a/src/llmcompressor/utils/transformers.py b/src/llmcompressor/utils/transformers.py index b8e61f725e..093b95dbad 100644 --- a/src/llmcompressor/utils/transformers.py +++ b/src/llmcompressor/utils/transformers.py @@ -79,7 +79,7 @@ def get_embeddings( try: input_embed = model.get_input_embeddings() - except (AttributeError, NotImplementedError): + except (AttributeError, NotImplementedError, TypeError): input_embed = None try: