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

Sign up or log in to comment