Skip to content

Latest commit

 

History

History
35 lines (27 loc) · 1.28 KB

README.md

File metadata and controls

35 lines (27 loc) · 1.28 KB

ExllamaV2 Worker on Runpod Serverless

This is worker code which uses ExllamaV2 for inference on Runpod Serverless.

🌟 How to use

  1. Clone this repository
  2. build docker image
  3. push docker image to your docker registry
  4. deploy to Runpod Serverless

🏗️ build docker image

docker build -t <your docker registry>/<your docker image name>:<your docker image tag> .

These are the build arguments:

Note: The model gets downloaded on first run so these models get downloaded on runtime instead of on build. Make sure you are attaching network volume to serverless endpoint so download only happens once and the cache is used for subsequent runs

key value optional
HUGGING_FACE_HUB_TOKEN your huggingface token true
MODEL_NAME your model name false
MODEL_REVISION your model revision true
MODEL_BASE_PATH your model base path true
LORA_ADAPTER_NAME your lora adapter name true
LORA_ADAPTER_REVISION your lora adapter revision true

⏫ push docker image to your docker registry

docker push <your docker registry>/<your docker image name>:<your docker image tag>

🚀 deploy to Runpod Serverless

After having docker image on your docker registry, you can deploy to Runpod Serverless.