Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity
A Gradio web UI for Large Language Models. Supports transformers, GPTQ, AWQ, EXL2, llama.cpp (GGUF), Llama models.
24 Commits 2 Branches 52 Tags 67 MiB
Python 90.6%
CSS 3.3%
JavaScript 3.1%
Shell 1.3%
Batchfile 0.7%
Other 1%
ee88d02292
Go to file
oobabooga ee88d02292 Small fix
2023-01-06 02:13:48 -03:00
presets Add support for presets 2023-01-06 01:33:21 -03:00
convert-to-torch.py Add file 2022-12-21 13:28:19 -03:00
LICENSE Initial commit 2022-12-21 01:17:38 -03:00
README.md Small fix 2023-01-06 02:13:48 -03:00
requirements.txt Add the requirements.txt file (duh) 2023-01-06 00:25:33 -03:00
server.py Autodetect available models 2023-01-06 02:06:59 -03:00
webui.png Add files via upload 2022-12-21 14:04:15 -03:00

README.md

text-generation-webui

A gradio webui for running large language models locally. Supports gpt-j-6B, gpt-neox-20b, opt, galactica, and many others.

Its goal is to become the AUTOMATIC1111/stable-diffusion-webui of text generation.

webui screenshot

Installation

Create a conda environment:

conda create -n textgen
conda activate textgen

Install the appropriate pytorch for your GPU. For NVIDIA GPUs, this should work:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

Install the requirements:

pip install -r requirements.txt

Downloading models

Models should be placed under models/model-name. For instance, models/gpt-j-6B for gpt-j-6B.

Hugging Face

Hugging Face is the main place to download models. These are some of my favorite:

The files that you need to download are the json, txt, and pytorch*.bin files. The remaining files are not necessary.

GPT-4chan

GPT-4chan has been shut down from Hugging Face, so you need to download it elsewhere. You have two options:

Converting to pytorch

The script convert-to-torch.py allows you to convert models to .pt format, which is about 10x faster to load:

python convert-to-torch.py models/model-name/

The output model will be saved to torch-dumps/model-name.pt. When you load a new model, the webui first looks for this .pt file; if it is not found, it loads the model as usual from models/model-name/.

Starting the webui

conda activate textgen
python server.py

Then browse to http://localhost:7860/?__theme=dark

Presets

Inference settings presets can be created under presets/ as text files. These files are detected automatically at startup.

Contributing

Pull requests are welcome.