Prepull model data on private swarm about petals

[poedator]

Hi, @wolfganghuse,
you can use snapshot_download:

from huggingface_hub import snapshot_download
snapshot_download(repo_id='petals-team/StableBeluga2')

The model files will appear in your standard HF cache location.

more details in HF Hub docs: https://huggingface.co/docs/huggingface_hub/en/guides/download

[wolfganghuse]

@poedator works exactly like you described...Thanks!

[wolfganghuse]

Thanks @poedator I will try as soon as possible!

[wolfganghuse]

Just to clarify: I would download into petals CACHE-DIR and petals would pick up the model and creates the block-links accordingly?

[poedator]

Just to clarify: I would download into petals CACHE-DIR and petals would pick up the model and creates the block-links accordingly?

Not exactly: It will download into HF cache dir, you have to move the cached model to the Petals cache manually. Pls try and see if it works.