model_head.pkl not loaded when HF repo is made private

[Ioannis-Pikoulis]

Hello.

Whenever I use SetFitModel.from_pretrained(model_id, use_auth_token=...) to load a model from a private repo, the model_head.pkl is not found. When the repo is made public, the head is found and loaded as normal. My repo contains both model_head.pkl and pytorch_model.bin files.

Please check this please.

[alvarobartt]

Hi @Ioannis-Pikoulis, please see @tomaarsen's comment at https://github.com/huggingface/setfit/issues/409#issuecomment-1683762555

[tomaarsen]

Hello!

I'm unable to reproduce this, I'm afraid. The use_auth_token is also correctly passed to the hf_hub_download method that is responsible for downloading this file.

• Tom Aarsen