TheBloke's Docker templates
Update: 16 December 2023 - Rebuild to add Mixtral support
Should now support Mixtral, with updated AutoGPTQ 0.6 and llama-cpp-python 0.2.23
Updated PyTorch to 2.1.1
Update: 11 October 2023 - Update API command line option
Container will now launch text-generation-webui with arg --extensions openai
Logs from text-generation-webui will now appear in the Runpod log viewer, as well as /workspace/logs/text-generation-webui.log
Update: 8th October 2023 - CUDA 12.1.1, fixed ExLlamav2 issues
The instances now use CUDA 12.1.1, which fixes issues with EXL2
Note that for now the main container is still called cuda11.8.0-ubuntu22.04-oneclick
This is because I need to get in touch with Runpod to update the name of the container used in their instances
This is just a naming issue; the container does now use CUDA 12.1.1 and EXL2 is confirmed to work again.
Update: 23rd July 2023 - Llama 2 support, including Llama 2 70B in ExLlama
Llama 2 models, including Llama 2 70B, are now fully supported
Updated to latest text-generation-webui requirements.txt
Removed the exllama pip package installed by text-generation-webui
Therefore the ExLlama kernel will build automatically on first use
This ensures that ExLlama is always up-to-date with any new ExLlama commits (which are pulled automatically on each boot)
Added simple build script for building the Docker containers
Update: 28th June 2023 - SuperHOT fixed
Updated to latest ExLlama code, fixing issue with SuperHOT GPTQs
ExLlama now automaticaly updates on boot, like text-generation-webui already did
This should result in the template automatically supporting new ExLlama features in future
Major update to the template
text-generation-webui is now integrated with:
AutoGPTQ with support for all Runpod GPU types
ExLlama, turbo-charged Llama GPTQ engine - performs 2x faster than AutoGPTQ (Llama 4bit GPTQs only)
CUDA-accelerated GGML support, with support for all Runpod systems and GPUs.
All text-generation-webui extensions are included and supported (Chat, SuperBooga, Whisper, etc).
text-generation-webui is always up-to-date with the latest code and features.
Automatic model download and loading via environment variable MODEL
.
Pass text-generation-webui parameters via environment variable UI_ARGS
.
Runpod: TheBloke's Local LLMs UI
Runpod template link
Full documentation is available here
Runpod: TheBloke's Local LLMs UI & API
Runpod template link
Full documentation is available here