Unable to download the model

#2
by sriharshasurineni - opened

I get the error -
ValueError: Could not load model OpenSafetyLab/MD-Judge-v0_2-internlm2_7b with any of the following classes: (<class 'transformers.models.auto.modeling_auto.AutoModelForCausalLM'>, <class 'transformers.models.auto.modeling_tf_auto.TFAutoModelForCausalLM'>). See the original errors:

while loading the model

OpenSafetyLab org
edited Oct 15, 2024

It would be helpful if you give me the code snippet since it looks like not an error from model_config

from transformers import pipeline

messages = [
{"role": "user", "content": "Who are you?"},
]
pipe = pipeline("text-generation", model="OpenSafetyLab/MD-Judge-v0_2-internlm2_7b", trust_remote_code=True)
pipe(messages)

OpenSafetyLab org

when I load model from local path, it's fine, maybe you should check your dependencies or simply load from local path after downloading

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment