Massaki Archambault 91d421848d | ||
---|---|---|
litellm | ||
.gitignore | ||
README.md | ||
docker-compose.amd.yml | ||
docker-compose.base.yml | ||
docker-compose.cpu.yml | ||
docker-compose.nvidia.yml |
README.md
local-llm
A quick prototype to self-host Open WebUI backed by Ollama to run LLM inference locally.
Getting started
Prerequisites
- Linux or WSL2
- docker
Steps for NVIDIA GPU
- Make sure your drivers are up to date.
- Install the NVIDIA Container Toolkit.
- Clone the repo.
- Symlink the NVIDIA compose spec to select it.
ln -s docker-compose.nvidia.yml docker.compose.yml
- Run
docker compose up
. Wait for a few minutes for the model to be downloaded and served. - Browse http://localhost:8080/
- Create an account and start chatting!
Steps for AMD GPU
Warning: AMD will doesn't support Windows at the moment. Use Linux.
- Make sure your drivers are up to date.
- Clone the repo.
- Symlink the AMD compose spec to select it.
ln -s docker-compose.amd.yml docker.compose.yml
- Run
docker compose up
. Wait for a few minutes for the model to be downloaded and served. - Browse http://localhost:8080/
- Create an account and start chatting!
Steps for NO GPU (use CPU)
Warning: This may be very slow depending on your CPU and may us a lot of RAM depending on the model
- Make sure your drivers are up to date.
- Clone the repo.
- Symlink the CPU compose spec to select it.
ln -s docker-compose.cpu.yml docker.compose.yml
- Run
docker compose up
. Wait for a few minutes for the model to be downloaded and served. - Browse http://localhost:8080/
- Create an account and start chatting!
Using the API
Open WebUI act as a proxy to Ollama and LiteLLM. For both API, authentication is done though a JWT token which can be fetched in the Settings > About page in Open WebUI.
Open WebUI exposes the Ollama API at the url http://localhost:8080/ollama/api.
Example usage:
curl -H "Authorization: Bearer <Paste your JWT token here>" http://localhost:8080/ollama/api/tags
The Ollama API can also be queried directly on port 11434, without proxing through Open WebUI. In some cases, like when working locally, it may be easier to use without having to proxy through Open WebUI. In that case, there is no authentification.
Example usage:
curl http://localhost:11434/api/tags
Ollama also have some OpenAI-compatible APIs. See the blog post for more detailed usage instructions.
Example usage:
curl http://localhost:11434/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "mistral",
"messages": [
{
"role": "system",
"content": "You are a helpful assistant."
},
{
"role": "user",
"content": "Hello!"
}
]
}'
Open WebUI exposes the LiteLLM API (for external providers) at the url http://localhost:8080/litellm/api/v1.
Example usage:
curl -H "Authorization: Bearer <Paste your JWT token here>" http://localhost:8080/litellm/api/v1/models
The JWT token can be used in place of the OpenAI API key for OpenAI-compatible libraries/applications.
Update
Simply run docker compose pull
followed by docker compose restart
.
Alternatives
Check out LM Studio for a more integrated, but non web-based alternative!