|
These are the VRAM and RAM requirements (in MiB) to run some examples of models **in 16-bit (default) precision**: |
|
|
|
| model | VRAM (GPU) | RAM | |
|
|:-----------------------|-------------:|--------:| |
|
| arxiv_ai_gpt2 | 1512.37 | 5824.2 | |
|
| blenderbot-1B-distill | 2441.75 | 4425.91 | |
|
| opt-1.3b | 2509.61 | 4427.79 | |
|
| gpt-neo-1.3b | 2605.27 | 5851.58 | |
|
| opt-2.7b | 5058.05 | 4863.95 | |
|
| gpt4chan_model_float16 | 11653.7 | 4437.71 | |
|
| gpt-j-6B | 11653.7 | 5633.79 | |
|
| galactica-6.7b | 12697.9 | 4429.89 | |
|
| opt-6.7b | 12700 | 4368.66 | |
|
| bloomz-7b1-p3 | 13483.1 | 4470.34 | |
|
|
|
#### GPU mode with 8-bit precision |
|
|
|
Allows you to load models that would not normally fit into your GPU. Enabled by default for 13b and 20b models in this web UI. |
|
|
|
| model | VRAM (GPU) | RAM | |
|
|:---------------|-------------:|--------:| |
|
| opt-13b | 12528.1 | 1152.39 | |
|
| gpt-neox-20b | 20384 | 2291.7 | |
|
|
|
#### CPU mode (32-bit precision) |
|
|
|
A lot slower, but does not require a GPU. |
|
|
|
On my i5-12400F, 6B models take around 10-20 seconds to respond in chat mode, and around 5 minutes to generate a 200 tokens completion. |
|
|
|
| model | RAM | |
|
|:-----------------------|---------:| |
|
| arxiv_ai_gpt2 | 4430.82 | |
|
| gpt-neo-1.3b | 6089.31 | |
|
| opt-1.3b | 8411.12 | |
|
| blenderbot-1B-distill | 8508.16 | |
|
| opt-2.7b | 14969.3 | |
|
| bloomz-7b1-p3 | 21371.2 | |
|
| gpt-j-6B | 24200.3 | |
|
| gpt4chan_model | 24246.3 | |
|
| galactica-6.7b | 26561.4 | |
|
| opt-6.7b | 29596.6 | |
|
|