Load models downloaded locally

#49
by buscon - opened

I am offering a workshop with stable audio open and I would like to offer the participants the option to load the model(s) from a local folder without using huggingface.

If I understood correctly, stable_audio_tools does not expose the real model classes for external imports, so this is not really possible.
Is this correct or is there a way to let python use a stable audio model on a specific path?

Usually you can do this with transformers, but it does not work with stable audio, because the models have a different structure

from transformers import AutoModel

model = AutoModel.from_pretrained('./bart-base')
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment