Add birefnet model_type and tag

#22
by Xenova HF staff - opened

Allows it to be searchable with https://huggingface.co/models?other=birefnet

OriLib changed pull request status to merged
This comment has been hidden
BRIA AI org

@Xenova i'm having problems due to this update:
"You are using a model of type birefnet to instantiate a model of type SegformerForSemanticSegmentation. This is not supported for all configurations of models and can yield errors."

also, when saving locally and loading i'm getting the following:
"Unrecognized model in rmbg/1/rmbg. Should have a model_type key in its config.json, or contain one of the following strings in its name: "

when adding model_type to config.json i get:
"The checkpoint you are trying to load has model type birefnet but Transformers does not recognize this architecture."

do you have a solution? should i revert this PR?

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment