https://github.com/dominodatalab/hf-modelregistry-examples

Integration supporting registration of LLM to Model Registry
https://github.com/dominodatalab/hf-modelregistry-examples

Last synced: 29 days ago
JSON representation

Integration supporting registration of LLM to Model Registry

• Host: GitHub
• URL: https://github.com/dominodatalab/hf-modelregistry-examples
• Owner: dominodatalab
• Created: 2024-10-17T10:55:45.000Z (2 months ago)
• Default Branch: main
• Last Pushed: 2024-10-17T13:44:13.000Z (2 months ago)
• Last Synced: 2024-10-19T19:05:45.306Z (2 months ago)
• Language: Jupyter Notebook
• Size: 11.7 KB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
## Registering Hugging Face Models via Domino Datasets

This repo demonstrates how Hugging Face models can be registered via Domino Datasets

The example notebook demonstrates how to register as well as run the models 

TBD - Running the registered model as a model api endpoint

## Run the model invocation as a job

Run the invocation job as follows

```

python /mtn/code/model_invocation_job.py --model_name "distilbert-base-uncased-finetuned-sst-2-english" --model_version 6 --text_input "I liked the movie"

```