apple

/

DCLM-7B

like 824

Follow

Apple 2,032

I am trying to load the model "apple/DCLM-7B" using the transformers library, but I am encountering a KeyError: 'openlm' when attempting to load the model. It seems that the model type openlm is not recognized by the current version of the transformers library.

Environment

• Transformers and Tokenizer versions: transformers-4.43.3 tokenizers-0.19.1
• Platform: Python 3 Google Compute Engine backend (GPU)
• Hardware accelerator: A100 GPU
• Python version: Python 3.10.12

Code Snippet

Here is the code I used to load the model:

# Importing necessary libraries from Hugging Face Transformers
from transformers import AutoModel, AutoTokenizer

# Specifying the model name
model_name = "apple/DCLM-7B"

# Loading the pre-trained model
model = AutoModel.from_pretrained(model_name)

# Loading the corresponding tokenizer
tokenizer = AutoTokenizer.from_pretrained(model_name)

# Example input text
input_text = "Hello, world!"

# Tokenizing the input text
inputs = tokenizer(input_text, return_tensors="pt")

# Getting the model's output
outputs = model(**inputs)

# Print the output
print(outputs)

Error Traceback

Here is the error traceback I received:

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
/usr/local/lib/python3.10/dist-packages/transformers/models/auto/configuration_auto.py in from_pretrained(cls, pretrained_model_name_or_path, **kwargs)
    981         fn.__doc__ = docstrings
--> 982         return fn
    983 

3 frames
KeyError: 'openlm'

During handling of the above exception, another exception occurred:

ValueError                                Traceback (most recent call last)
/usr/local/lib/python3.10/dist-packages/transformers/models/auto/configuration_auto.py in from_pretrained(cls, pretrained_model_name_or_path, **kwargs)
    982         return fn
    983 
--> 984     return docstring_decorator
    985 
    986 

ValueError: The checkpoint you are trying to load has model type `openlm` but Transformers does not recognize this architecture. 

This could be because of an issue with the checkpoint, or because your version of Transformers is out of date.

Steps to Reproduce

1. Install the transformers library: pip install transformers
2. Run the code snippet provided above to load the model apple/DCLM-7B

Additional Context

I have also tried updating the transformers library to the latest version using pip install --upgrade transformers, but the issue persists.