Jupyter Notebook container image for running and fine-tuning Large Language Models
- rclone
- FastChat
- Ollama
- VS Code Server
- Jupyter AI
- bitsandbytes
- transformers
- peft
- accelerate
- trl
- ollama-python
- openai
- pyaudio
- portaudio
- cuda-nvcc
- deepspeed
- langchain
- huggingface_hub
- gptqmodel
- autoawq
- xformers
- dask-kubernetes
- chromadb
This image is based upon the Jupyter Stack PyTorch image