lxtGH / OMG-Seg

OMG-LLaVA and OMG-Seg codebase
Other
882 stars 43 forks source link

Huggingface loading is not valid #12

Open Yonggie opened 4 days ago

Yonggie commented 4 days ago

https://hf-mirror.com/LXT/OMG_Seg/tree/main

from transformers import AutoModel

AutoModel.from_pretrained(OMG-Seg)

got error no config.json 图片

Yonggie commented 4 days ago

Or could u kindly gives a simple easy example of loading the pretrained model? Something like:

# load_img
# load model

res=model.predict(img)
lxtGH commented 4 days ago

@Yonggie Hi, Please wait for the new update for easier use for demo in this weekend.

Yonggie commented 4 days ago

I guess this is just uploading the checkpoint(pth) to hugging face instead of making the model into huggingface. I mean it's also fine, but it's hard to load the pth as no examples provided in the name card...

THX btw, excellent work! 图片