Hey guys,
In the current API, unless I am doing something wrong, I have to specify each remote file (config, PyTorch model) explicitly in order to load a custom model outside of HF repo ecosystem.
For example, to load custom model private_model_name from a hypothetical remote repo example.com I need to do the following:
config = AutoConfig.from_pretrained(“https://example.com/models/private_model_name/config.json”)
model = AutoModel.from_pretrained(“https://example.com/models/private_model_name/pytorch_model.bin”, config=config)
Why not add naming resolution capabilities that currently exist to all remote repos so we can do:
model = AutoModel.from_pretrained(“https://example.com/models/private_model_name/”)
Can we add the same name resolution assumption for non-HF model repos?