huggingface / huggingface/setfit

model_head.pkl not loaded when HF repo is made private

Open
#415 2 comments 0 reactions 0 assignees View on GitHub
needs verification
Dominant language
Jupyter Notebook
Stars
2.8k
Forks
267
Avg merge
36m
Merged PRs (30d)
5

Description

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.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.