Error loading fine tuned model

#5
by codegood - opened

I fine-tuned the Microsoft/Florence-2-base on my task. I can't load my model again due to the following error: AssertionError: only DaViT is supported for now. I see that you could pass through this error using a custom loading script.

Please guide me on how I can do the same. Thanks

codegood changed discussion title from Issue when fine tuning to Error loading fine tuned model
HuggingFaceM4 org

Edit the config.json file. One of the model_type keys will have an empty value (""). Replace that with "davit"

andito changed discussion status to closed

Sign up or log in to comment