Error when using model

#2
by wardaslab - opened

I am trying to extract features from my videos using the code below:

model = AutoModel.from_pretrained(
MODEL_NAME,
trust_remote_code=True,
token=HF_TOKEN # Only needed if model is gated
).eval().to(device)

But I am getting this error:
ValueError: Unrecognized model in OpenGVLab/InternVideo2-Stage2_1B-224p-f4. Should have a model_type key in its config.json, or contain one of the following strings in its name: afmoe, aimv2, aimv2_vision_model, albert, align, altclip, apertus, arcee, aria, aria_text, audio-spectrogram-transformer...

OpenGVLab org

There is no a config file, please use an example in repo, you can load a model and extract vision features

Sign up or log in to comment