text-generation-webui/docs/LLaMA-model.md

52 lines
1.9 KiB
Markdown
Raw Normal View History

2023-04-22 01:34:13 -04:00
LLaMA is a Large Language Model developed by Meta AI.
It was trained on more tokens than previous models. The result is that the smallest version with 7 billion parameters has similar performance to GPT-3 with 175 billion parameters.
This guide will cover usage through the official `transformers` implementation. For 4-bit mode, head over to [GPTQ models (4 bit mode)
2023-04-22 01:49:54 -04:00
](GPTQ-models-(4-bit-mode).md).
2023-04-22 01:34:13 -04:00
## Getting the weights
### Option 1: pre-converted weights
* Torrent: https://github.com/oobabooga/text-generation-webui/pull/530#issuecomment-1484235789
* Direct download: https://huggingface.co/Neko-Institute-of-Science
2023-04-25 20:28:15 -04:00
⚠️ The tokenizers for the Torrent source above and also for many LLaMA fine-tunes available on Hugging Face may be outdated, so I recommend downloading the following universal LLaMA tokenizer:
2023-04-22 01:34:13 -04:00
```
python download-model.py oobabooga/llama-tokenizer
```
Once downloaded, it will be automatically applied to **every** `LlamaForCausalLM` model that you try to load.
### Option 2: convert the weights yourself
1. Install the `protobuf` library:
```
pip install protobuf==3.20.1
2023-04-22 01:34:13 -04:00
```
2023-06-08 23:30:22 -04:00
2. Use the script below to convert the model in `.pth` format that you, a fellow academic, downloaded using Meta's official link.
2023-04-22 01:34:13 -04:00
2023-06-08 23:30:22 -04:00
If you have `transformers` installed in place:
```
python -m transformers.models.llama.convert_llama_weights_to_hf --input_dir /path/to/LLaMA --model_size 7B --output_dir /tmp/outputs/llama-7b
```
2023-06-08 23:30:22 -04:00
Otherwise download [convert_llama_weights_to_hf.py](https://github.com/huggingface/transformers/blob/main/src/transformers/models/llama/convert_llama_weights_to_hf.py) first and run:
2023-04-22 01:34:13 -04:00
```
python convert_llama_weights_to_hf.py --input_dir /path/to/LLaMA --model_size 7B --output_dir /tmp/outputs/llama-7b
```
3. Move the `llama-7b` folder inside your `text-generation-webui/models` folder.
## Starting the web UI
```python
python server.py --model llama-7b
2023-04-22 01:49:54 -04:00
```