Integrate to AutoModel

#10
by Samoed - opened

Integrate gme model with Transformers AutoModel.
Now it's available with

import torch
model = AutoModel.from_pretrained("./gme-Qwen2-VL-2B-Instruct", trust_remote_code=True)

@thenlper @izhx can you review?

Samoed changed pull request status to open
Samoed changed pull request title from Try to integrate AutoModel to Integrate to AutoModel
Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment