Converted the model to safetensors
Hello Mentors,
I have been working on converting the model to the SafeTensors format and have set num_classes = 2. I would appreciate any guidance or clarification on whether this approach is correct.
Thank you for your time and support.
Thanks for this
@Rhydham
and apologies for the delay. Two classes should be correct. As part of this conversion we also need a config.json file added to the repo. These crop models are a little different, so I'm going to tag in
@bw4sz
and
@Henrykironde
in case they have any guidance on creating that file. For the non-crop models you can create it using model.svae_pretrained() (https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.ModelHubMixin.save_pretrained) once the model is properly loaded, but I'm not sure if we have it setup to work that way for the crop models.