Skip to content

vikalluru/nemo-embedding-demo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

source env_variables.txt

ngc registry model download-version "ohlfw0olaadg/ea-participants/nv-embed-qa-mistral-7b:1" chmod -R o+rX nv-embed-qa-mistral-7b_v1

docker pull nvcr.io/ohlfw0olaadg/ea-participants/nemo-retriever-embedding-microservice:24.04

docker run
-it
--name $CONTAINER_NAME
--gpus=1
--shm-size=8G
-v $(pwd):/model-checkpoint-path
-p $SERVICE_PORT:$SERVICE_PORT
nvcr.io/ohlfw0olaadg/ea-participants/nemo-retriever-embedding-microservice:24.04
bin/web -p $SERVICE_PORT -c /model-checkpoint-path/$MODEL_NAME -g model_config_templates/${MODEL_ID}_template.yaml

curl -X "POST" "http://51.124.97.12:8080/v1/embeddings" -H 'accept: application/json' -H 'Content-Type: application/json' -d '{ "input": ["Hello"], "model": "NV-Embed-QA-Mistral-7B", "input_type": "query" }'

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published