incorrect import
thank you for bringing this to the platform so quickly!
maybe this should be a pull request, but the current model card/readme doesn't work as intended. specifically, build_sam only works with vit_h, whereas the given ckpt is for vit_b. i believe it's best to route build_sam based on the given ckpt, like how automodel.from_pretrained(..) etc operates.
hi
@ozanciga
As SAM is part of transformers, I would advise to refer to the HF official docs here: https://huggingface.co./docs/transformers/main/en/model_doc/sam as we currently maintain that version rather than this repo
@ybelkada if I upload a fine tuned sam version, how does the api know my outputs? and input? where do I host the code? Can you shed some light on converting it to an API?