TheBloke commited on
Commit
6102e5d
1 Parent(s): 1cc34be

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +63 -48
README.md CHANGED
@@ -29,18 +29,24 @@ quantized_by: TheBloke
29
  - Model creator: [OpenAssistant](https://huggingface.co/OpenAssistant)
30
  - Original model: [Llama2 13B MegaCode2 OASST](https://huggingface.co/OpenAssistant/llama2-13b-megacode2-oasst)
31
 
 
32
  ## Description
33
 
34
  This repo contains GPTQ model files for [OpenAssistant's Llama2 13B MegaCode2 OASST](https://huggingface.co/OpenAssistant/llama2-13b-megacode2-oasst).
35
 
36
  Multiple GPTQ parameter permutations are provided; see Provided Files below for details of the options provided, their parameters, and the software used to create them.
37
 
 
 
38
  ## Repositories available
39
 
40
  * [GPTQ models for GPU inference, with multiple quantisation parameter options.](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ)
41
- * [2, 3, 4, 5, 6 and 8-bit GGML models for CPU+GPU inference](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GGML)
 
42
  * [OpenAssistant's original unquantised fp16 model in pytorch format, for GPU inference and for further conversions](https://huggingface.co/OpenAssistant/llama2-13b-megacode2-oasst)
 
43
 
 
44
  ## Prompt template: ChatML
45
 
46
  ```
@@ -49,22 +55,26 @@ Multiple GPTQ parameter permutations are provided; see Provided Files below for
49
  <|im_start|>user
50
  {prompt}<|im_end|>
51
  <|im_start|>assistant
 
52
  ```
53
 
 
 
 
54
  ## Provided files and GPTQ parameters
55
 
56
  Multiple quantisation parameters are provided, to allow you to choose the best one for your hardware and requirements.
57
 
58
  Each separate quant is in a different branch. See below for instructions on fetching from different branches.
59
 
60
- All GPTQ files are made with AutoGPTQ.
61
 
62
  <details>
63
  <summary>Explanation of GPTQ parameters</summary>
64
 
65
  - Bits: The bit size of the quantised model.
66
  - GS: GPTQ group size. Higher numbers use less VRAM, but have lower quantisation accuracy. "None" is the lowest possible value.
67
- - Act Order: True or False. Also known as `desc_act`. True results in better quantisation accuracy. Some GPTQ clients have issues with models that use Act Order plus Group Size.
68
  - Damp %: A GPTQ parameter that affects how samples are processed for quantisation. 0.01 is default, but 0.1 results in slightly better accuracy.
69
  - GPTQ dataset: The dataset used for quantisation. Using a dataset more appropriate to the model's training can improve quantisation accuracy. Note that the GPTQ dataset is not the same as the dataset used to train the model - please refer to the original model repo for details of the training dataset(s).
70
  - Sequence Length: The length of the dataset sequences used for quantisation. Ideally this is the same as the model sequence length. For some very long sequence models (16+K), a lower sequence length may have to be used. Note that a lower sequence length does not limit the sequence length of the quantised model. It only impacts the quantisation accuracy on longer inference sequences.
@@ -74,13 +84,16 @@ All GPTQ files are made with AutoGPTQ.
74
 
75
  | Branch | Bits | GS | Act Order | Damp % | GPTQ Dataset | Seq Len | Size | ExLlama | Desc |
76
  | ------ | ---- | -- | --------- | ------ | ------------ | ------- | ---- | ------- | ---- |
77
- | [main](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/main) | 4 | 128 | No | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 7.26 GB | Yes | Most compatible option. Good inference speed in AutoGPTQ and GPTQ-for-LLaMa. Lower inference quality than other options. |
78
- | [gptq-4bit-32g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-4bit-32g-actorder_True) | 4 | 32 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 8.00 GB | Yes | 4-bit, with Act Order and group size 32g. Gives highest possible inference quality, with maximum VRAM usage. Poor AutoGPTQ CUDA speed. |
79
- | [gptq-4bit-64g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-4bit-64g-actorder_True) | 4 | 64 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 7.51 GB | Yes | 4-bit, with Act Order and group size 64g. Uses less VRAM than 32g, but with slightly lower accuracy. Poor AutoGPTQ CUDA speed. |
80
- | [gptq-4bit-128g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-4bit-128g-actorder_True) | 4 | 128 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 7.26 GB | Yes | 4-bit, with Act Order and group size 128g. Uses even less VRAM than 64g, but with slightly lower accuracy. Poor AutoGPTQ CUDA speed. |
81
- | [gptq-8bit--1g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-8bit--1g-actorder_True) | 8 | None | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 13.36 GB | No | 8-bit, with Act Order. No group size, to lower VRAM requirements and to improve AutoGPTQ speed. |
82
  | [gptq-8bit-128g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-8bit-128g-actorder_True) | 8 | 128 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 13.65 GB | No | 8-bit, with group size 128g for higher inference quality and with Act Order for even higher accuracy. Poor AutoGPTQ CUDA speed. |
83
 
 
 
 
84
  ## How to download from branches
85
 
86
  - In text-generation-webui, you can add `:branch` to the end of the download name, eg `TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ:gptq-4bit-32g-actorder_True`
@@ -89,79 +102,79 @@ All GPTQ files are made with AutoGPTQ.
89
  git clone --single-branch --branch gptq-4bit-32g-actorder_True https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ
90
  ```
91
  - In Python Transformers code, the branch is the `revision` parameter; see below.
92
-
 
93
  ## How to easily download and use this model in [text-generation-webui](https://github.com/oobabooga/text-generation-webui).
94
 
95
  Please make sure you're using the latest version of [text-generation-webui](https://github.com/oobabooga/text-generation-webui).
96
 
97
- It is strongly recommended to use the text-generation-webui one-click-installers unless you know how to make a manual install.
98
 
99
  1. Click the **Model tab**.
100
  2. Under **Download custom model or LoRA**, enter `TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ`.
101
  - To download from a specific branch, enter for example `TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ:gptq-4bit-32g-actorder_True`
102
  - see Provided Files above for the list of branches for each option.
103
  3. Click **Download**.
104
- 4. The model will start downloading. Once it's finished it will say "Done"
105
  5. In the top left, click the refresh icon next to **Model**.
106
  6. In the **Model** dropdown, choose the model you just downloaded: `Llama2-13B-MegaCode2-OASST-GPTQ`
107
  7. The model will automatically load, and is now ready for use!
108
  8. If you want any custom settings, set them and then click **Save settings for this model** followed by **Reload the Model** in the top right.
109
- * Note that you do not need to set GPTQ parameters any more. These are set automatically from the file `quantize_config.json`.
110
  9. Once you're ready, click the **Text Generation tab** and enter a prompt to get started!
 
111
 
 
112
  ## How to use this GPTQ model from Python code
113
 
114
- First make sure you have [AutoGPTQ](https://github.com/PanQiWei/AutoGPTQ) 0.3.1 or later installed:
115
 
116
- ```
117
- pip3 install auto-gptq
118
- ```
119
 
120
- If you have problems installing AutoGPTQ, please build from source instead:
 
 
121
  ```
 
 
 
 
122
  pip3 uninstall -y auto-gptq
123
  git clone https://github.com/PanQiWei/AutoGPTQ
124
  cd AutoGPTQ
125
  pip3 install .
126
  ```
127
 
128
- Then try the following example code:
 
 
 
 
 
 
 
 
129
 
130
  ```python
131
- from transformers import AutoTokenizer, pipeline, logging
132
- from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
133
 
134
  model_name_or_path = "TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ"
135
-
136
- use_triton = False
 
 
 
 
137
 
138
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
139
 
140
- model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
141
- use_safetensors=True,
142
- trust_remote_code=False,
143
- device="cuda:0",
144
- use_triton=use_triton,
145
- quantize_config=None)
146
-
147
- """
148
- # To download from a specific branch, use the revision parameter, as in this example:
149
- # Note that `revision` requires AutoGPTQ 0.3.1 or later!
150
-
151
- model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
152
- revision="gptq-4bit-32g-actorder_True",
153
- use_safetensors=True,
154
- trust_remote_code=False,
155
- device="cuda:0",
156
- quantize_config=None)
157
- """
158
-
159
  prompt = "Tell me about AI"
160
  prompt_template=f'''<|im_start|>system
161
  {system_message}<|im_end|>
162
  <|im_start|>user
163
  {prompt}<|im_end|>
164
  <|im_start|>assistant
 
165
  '''
166
 
167
  print("\n\n*** Generate:")
@@ -172,9 +185,6 @@ print(tokenizer.decode(output[0]))
172
 
173
  # Inference can also be done using transformers' pipeline
174
 
175
- # Prevent printing spurious transformers error when using pipeline with AutoGPTQ
176
- logging.set_verbosity(logging.CRITICAL)
177
-
178
  print("*** Pipeline:")
179
  pipe = pipeline(
180
  "text-generation",
@@ -188,12 +198,17 @@ pipe = pipeline(
188
 
189
  print(pipe(prompt_template)[0]['generated_text'])
190
  ```
 
191
 
 
192
  ## Compatibility
193
 
194
- The files provided will work with AutoGPTQ (CUDA and Triton modes), GPTQ-for-LLaMa (only CUDA has been tested), and Occ4m's GPTQ-for-LLaMa fork.
 
 
195
 
196
- ExLlama works with Llama models in 4-bit. Please see the Provided Files table above for per-file compatibility.
 
197
 
198
  <!-- footer start -->
199
  <!-- 200823 -->
@@ -218,7 +233,7 @@ Donaters will get priority support on any and all AI/LLM/model questions and req
218
 
219
  **Special thanks to**: Aemon Algiz.
220
 
221
- **Patreon special mentions**: Sam, theTransient, Jonathan Leane, Steven Wood, webtim, Johann-Peter Hartmann, Geoffrey Montalvo, Gabriel Tamborski, Willem Michiel, John Villwock, Derek Yates, Mesiah Bishop, Eugene Pentland, Pieter, Chadd, Stephen Murray, Daniel P. Andersen, terasurfer, Brandon Frisco, Thomas Belote, Sid, Nathan LeClaire, Magnesian, Alps Aficionado, Stanislav Ovsiannikov, Alex, Joseph William Delisle, Nikolai Manek, Michael Davis, Junyu Yang, K, J, Spencer Kim, Stefan Sabev, Olusegun Samson, transmissions 11, Michael Levine, Cory Kujawski, Rainer Wilmers, zynix, Kalila, Luke @flexchar, Ajan Kanaga, Mandus, vamX, Ai Maven, Mano Prime, Matthew Berman, subjectnull, Vitor Caleffi, Clay Pascal, biorpg, alfie_i, 阿明, Jeffrey Morgan, ya boyyy, Raymond Fosdick, knownsqashed, Olakabola, Leonard Tan, ReadyPlayerEmma, Enrico Ros, Dave, Talal Aujan, Illia Dulskyi, Sean Connelly, senxiiz, Artur Olbinski, Elle, Raven Klaugh, Fen Risland, Deep Realms, Imad Khwaja, Fred von Graf, Will Dee, usrbinkat, SuperWojo, Alexandros Triantafyllidis, Swaroop Kallakuri, Dan Guido, John Detwiler, Pedro Madruga, Iucharbius, Viktor Bowallius, Asp the Wyvern, Edmond Seymore, Trenton Dambrowitz, Space Cruiser, Spiking Neurons AB, Pyrater, LangChain4j, Tony Hughes, Kacper Wikieł, Rishabh Srivastava, David Ziegler, Luke Pendergrass, Andrey, Gabriel Puliatti, Lone Striker, Sebastain Graf, Pierre Kircher, Randy H, NimbleBox.ai, Vadim, danny, Deo Leter
222
 
223
 
224
  Thank you to all my generous patrons and donaters!
@@ -236,7 +251,7 @@ And thank you again to a16z for their generous grant.
236
  ### Prompt template
237
 
238
  [chatml](https://github.com/openai/openai-python/blob/main/chatml.md) format is used:
239
- "<|im_start|>user\n{user prompt}<|im_end|>\n<|im_start|>assistant\n{Assistant answer}<|im_end|>\n"
240
 
241
  Multi-line:
242
 
 
29
  - Model creator: [OpenAssistant](https://huggingface.co/OpenAssistant)
30
  - Original model: [Llama2 13B MegaCode2 OASST](https://huggingface.co/OpenAssistant/llama2-13b-megacode2-oasst)
31
 
32
+ <!-- description start -->
33
  ## Description
34
 
35
  This repo contains GPTQ model files for [OpenAssistant's Llama2 13B MegaCode2 OASST](https://huggingface.co/OpenAssistant/llama2-13b-megacode2-oasst).
36
 
37
  Multiple GPTQ parameter permutations are provided; see Provided Files below for details of the options provided, their parameters, and the software used to create them.
38
 
39
+ <!-- description end -->
40
+ <!-- repositories-available start -->
41
  ## Repositories available
42
 
43
  * [GPTQ models for GPU inference, with multiple quantisation parameter options.](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ)
44
+ * [2, 3, 4, 5, 6 and 8-bit GGUF models for CPU+GPU inference](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GGUF)
45
+ * [2, 3, 4, 5, 6 and 8-bit GGML models for CPU+GPU inference (deprecated)](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GGML)
46
  * [OpenAssistant's original unquantised fp16 model in pytorch format, for GPU inference and for further conversions](https://huggingface.co/OpenAssistant/llama2-13b-megacode2-oasst)
47
+ <!-- repositories-available end -->
48
 
49
+ <!-- prompt-template start -->
50
  ## Prompt template: ChatML
51
 
52
  ```
 
55
  <|im_start|>user
56
  {prompt}<|im_end|>
57
  <|im_start|>assistant
58
+
59
  ```
60
 
61
+ <!-- prompt-template end -->
62
+
63
+ <!-- README_GPTQ.md-provided-files start -->
64
  ## Provided files and GPTQ parameters
65
 
66
  Multiple quantisation parameters are provided, to allow you to choose the best one for your hardware and requirements.
67
 
68
  Each separate quant is in a different branch. See below for instructions on fetching from different branches.
69
 
70
+ All recent GPTQ files are made with AutoGPTQ, and all files in non-main branches are made with AutoGPTQ. Files in the `main` branch which were uploaded before August 2023 were made with GPTQ-for-LLaMa.
71
 
72
  <details>
73
  <summary>Explanation of GPTQ parameters</summary>
74
 
75
  - Bits: The bit size of the quantised model.
76
  - GS: GPTQ group size. Higher numbers use less VRAM, but have lower quantisation accuracy. "None" is the lowest possible value.
77
+ - Act Order: True or False. Also known as `desc_act`. True results in better quantisation accuracy. Some GPTQ clients have had issues with models that use Act Order plus Group Size, but this is generally resolved now.
78
  - Damp %: A GPTQ parameter that affects how samples are processed for quantisation. 0.01 is default, but 0.1 results in slightly better accuracy.
79
  - GPTQ dataset: The dataset used for quantisation. Using a dataset more appropriate to the model's training can improve quantisation accuracy. Note that the GPTQ dataset is not the same as the dataset used to train the model - please refer to the original model repo for details of the training dataset(s).
80
  - Sequence Length: The length of the dataset sequences used for quantisation. Ideally this is the same as the model sequence length. For some very long sequence models (16+K), a lower sequence length may have to be used. Note that a lower sequence length does not limit the sequence length of the quantised model. It only impacts the quantisation accuracy on longer inference sequences.
 
84
 
85
  | Branch | Bits | GS | Act Order | Damp % | GPTQ Dataset | Seq Len | Size | ExLlama | Desc |
86
  | ------ | ---- | -- | --------- | ------ | ------------ | ------- | ---- | ------- | ---- |
87
+ | [main](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/main) | 4 | 128 | No | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 7.26 GB | Yes | Most compatible option. Good inference speed in AutoGPTQ and GPTQ-for-LLaMa. Lower inference quality than other options. |
88
+ | [gptq-4bit-32g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-4bit-32g-actorder_True) | 4 | 32 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 8.00 GB | Yes | 4-bit, with Act Order and group size 32g. Gives highest possible inference quality, with maximum VRAM usage. Poor AutoGPTQ CUDA speed. |
89
+ | [gptq-4bit-64g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-4bit-64g-actorder_True) | 4 | 64 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 7.51 GB | Yes | 4-bit, with Act Order and group size 64g. Uses less VRAM than 32g, but with slightly lower accuracy. Poor AutoGPTQ CUDA speed. |
90
+ | [gptq-4bit-128g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-4bit-128g-actorder_True) | 4 | 128 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 7.26 GB | Yes | 4-bit, with Act Order and group size 128g. Uses even less VRAM than 64g, but with slightly lower accuracy. Poor AutoGPTQ CUDA speed. |
91
+ | [gptq-8bit--1g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-8bit--1g-actorder_True) | 8 | None | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 13.36 GB | No | 8-bit, with Act Order. No group size, to lower VRAM requirements and to improve AutoGPTQ speed. |
92
  | [gptq-8bit-128g-actorder_True](https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ/tree/gptq-8bit-128g-actorder_True) | 8 | 128 | Yes | 0.1 | [Evol Instruct Code](https://huggingface.co/datasets/nickrosh/Evol-Instruct-Code-80k-v1) | 4096 | 13.65 GB | No | 8-bit, with group size 128g for higher inference quality and with Act Order for even higher accuracy. Poor AutoGPTQ CUDA speed. |
93
 
94
+ <!-- README_GPTQ.md-provided-files end -->
95
+
96
+ <!-- README_GPTQ.md-download-from-branches start -->
97
  ## How to download from branches
98
 
99
  - In text-generation-webui, you can add `:branch` to the end of the download name, eg `TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ:gptq-4bit-32g-actorder_True`
 
102
  git clone --single-branch --branch gptq-4bit-32g-actorder_True https://huggingface.co/TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ
103
  ```
104
  - In Python Transformers code, the branch is the `revision` parameter; see below.
105
+ <!-- README_GPTQ.md-download-from-branches end -->
106
+ <!-- README_GPTQ.md-text-generation-webui start -->
107
  ## How to easily download and use this model in [text-generation-webui](https://github.com/oobabooga/text-generation-webui).
108
 
109
  Please make sure you're using the latest version of [text-generation-webui](https://github.com/oobabooga/text-generation-webui).
110
 
111
+ It is strongly recommended to use the text-generation-webui one-click-installers unless you're sure you know how to make a manual install.
112
 
113
  1. Click the **Model tab**.
114
  2. Under **Download custom model or LoRA**, enter `TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ`.
115
  - To download from a specific branch, enter for example `TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ:gptq-4bit-32g-actorder_True`
116
  - see Provided Files above for the list of branches for each option.
117
  3. Click **Download**.
118
+ 4. The model will start downloading. Once it's finished it will say "Done".
119
  5. In the top left, click the refresh icon next to **Model**.
120
  6. In the **Model** dropdown, choose the model you just downloaded: `Llama2-13B-MegaCode2-OASST-GPTQ`
121
  7. The model will automatically load, and is now ready for use!
122
  8. If you want any custom settings, set them and then click **Save settings for this model** followed by **Reload the Model** in the top right.
123
+ * Note that you do not need to and should not set manual GPTQ parameters any more. These are set automatically from the file `quantize_config.json`.
124
  9. Once you're ready, click the **Text Generation tab** and enter a prompt to get started!
125
+ <!-- README_GPTQ.md-text-generation-webui end -->
126
 
127
+ <!-- README_GPTQ.md-use-from-python start -->
128
  ## How to use this GPTQ model from Python code
129
 
130
+ ### Install the necessary packages
131
 
132
+ Requires: Transformers 4.32.0 or later, Optimum 1.12.0 or later, and AutoGPTQ 0.4.2 or later.
 
 
133
 
134
+ ```shell
135
+ pip3 install transformers>=4.32.0 optimum>=1.12.0
136
+ pip3 install auto-gptq --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/ # Use cu117 if on CUDA 11.7
137
  ```
138
+
139
+ If you have problems installing AutoGPTQ using the pre-built wheels, install it from source instead:
140
+
141
+ ```shell
142
  pip3 uninstall -y auto-gptq
143
  git clone https://github.com/PanQiWei/AutoGPTQ
144
  cd AutoGPTQ
145
  pip3 install .
146
  ```
147
 
148
+ ### For CodeLlama models only: you must use Transformers 4.33.0 or later.
149
+
150
+ If 4.33.0 is not yet released when you read this, you will need to install Transformers from source:
151
+ ```shell
152
+ pip3 uninstall -y transformers
153
+ pip3 install git+https://github.com/huggingface/transformers.git
154
+ ```
155
+
156
+ ### You can then use the following code
157
 
158
  ```python
159
+ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
160
 
161
  model_name_or_path = "TheBloke/Llama2-13B-MegaCode2-OASST-GPTQ"
162
+ # To use a different branch, change revision
163
+ # For example: revision="gptq-4bit-32g-actorder_True"
164
+ model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
165
+ torch_dtype=torch.float16,
166
+ device_map="auto",
167
+ revision="main")
168
 
169
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
170
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
171
  prompt = "Tell me about AI"
172
  prompt_template=f'''<|im_start|>system
173
  {system_message}<|im_end|>
174
  <|im_start|>user
175
  {prompt}<|im_end|>
176
  <|im_start|>assistant
177
+
178
  '''
179
 
180
  print("\n\n*** Generate:")
 
185
 
186
  # Inference can also be done using transformers' pipeline
187
 
 
 
 
188
  print("*** Pipeline:")
189
  pipe = pipeline(
190
  "text-generation",
 
198
 
199
  print(pipe(prompt_template)[0]['generated_text'])
200
  ```
201
+ <!-- README_GPTQ.md-use-from-python end -->
202
 
203
+ <!-- README_GPTQ.md-compatibility start -->
204
  ## Compatibility
205
 
206
+ The files provided are tested to work with AutoGPTQ, both via Transformers and using AutoGPTQ directly. They should also work with [Occ4m's GPTQ-for-LLaMa fork](https://github.com/0cc4m/KoboldAI).
207
+
208
+ [ExLlama](https://github.com/turboderp/exllama) is compatible with Llama models in 4-bit. Please see the Provided Files table above for per-file compatibility.
209
 
210
+ [Huggingface Text Generation Inference (TGI)](https://github.com/huggingface/text-generation-inference) is compatible with all GPTQ models.
211
+ <!-- README_GPTQ.md-compatibility end -->
212
 
213
  <!-- footer start -->
214
  <!-- 200823 -->
 
233
 
234
  **Special thanks to**: Aemon Algiz.
235
 
236
+ **Patreon special mentions**: Russ Johnson, J, alfie_i, Alex, NimbleBox.ai, Chadd, Mandus, Nikolai Manek, Ken Nordquist, ya boyyy, Illia Dulskyi, Viktor Bowallius, vamX, Iucharbius, zynix, Magnesian, Clay Pascal, Pierre Kircher, Enrico Ros, Tony Hughes, Elle, Andrey, knownsqashed, Deep Realms, Jerry Meng, Lone Striker, Derek Yates, Pyrater, Mesiah Bishop, James Bentley, Femi Adebogun, Brandon Frisco, SuperWojo, Alps Aficionado, Michael Dempsey, Vitor Caleffi, Will Dee, Edmond Seymore, usrbinkat, LangChain4j, Kacper Wikieł, Luke Pendergrass, John Detwiler, theTransient, Nathan LeClaire, Tiffany J. Kim, biorpg, Eugene Pentland, Stanislav Ovsiannikov, Fred von Graf, terasurfer, Kalila, Dan Guido, Nitin Borwankar, 阿明, Ai Maven, John Villwock, Gabriel Puliatti, Stephen Murray, Asp the Wyvern, danny, Chris Smitley, ReadyPlayerEmma, S_X, Daniel P. Andersen, Olakabola, Jeffrey Morgan, Imad Khwaja, Caitlyn Gatomon, webtim, Alicia Loh, Trenton Dambrowitz, Swaroop Kallakuri, Erik Bjäreholt, Leonard Tan, Spiking Neurons AB, Luke @flexchar, Ajan Kanaga, Thomas Belote, Deo Leter, RoA, Willem Michiel, transmissions 11, subjectnull, Matthew Berman, Joseph William Delisle, David Ziegler, Michael Davis, Johann-Peter Hartmann, Talal Aujan, senxiiz, Artur Olbinski, Rainer Wilmers, Spencer Kim, Fen Risland, Cap'n Zoog, Rishabh Srivastava, Michael Levine, Geoffrey Montalvo, Sean Connelly, Alexandros Triantafyllidis, Pieter, Gabriel Tamborski, Sam, Subspace Studios, Junyu Yang, Pedro Madruga, Vadim, Cory Kujawski, K, Raven Klaugh, Randy H, Mano Prime, Sebastain Graf, Space Cruiser
237
 
238
 
239
  Thank you to all my generous patrons and donaters!
 
251
  ### Prompt template
252
 
253
  [chatml](https://github.com/openai/openai-python/blob/main/chatml.md) format is used:
254
+ "<|im_start|>user\n{user prompt}<|im_end|>\n<|im_start|>assistant\n{Assistant answer}<|im_end|>\n"
255
 
256
  Multi-line:
257