Skip to content

Converting model checkpoint(.pth) to HuggingFace Model #730

Answered by pascal-pfeiffer
sunilswain asked this question in Q&A
Discussion options

You must be logged in to vote

When using the UI, you can follow the steps here (https://h2oai.github.io/h2o-llmstudio/guide/experiments/export-trained-model) to push your model to Hugging Face Hub or to download locally.

When using the CLI, use the publish_to_huggingface.py script.

make shell
python publish_to_huggingface.py --path_to_experiment YOUR_PATH 

There are a few more args to fill if you need them. Show with:

python publish_to_huggingface.py --help

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by sunilswain
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants