Update README.md

README.md

@@ -23,6 +23,14 @@ These files are GPTQ 4bit model files for [Panchovix's merge of WizardLM 33B V1.
 
 It is the result of quantising to 4bit using [GPTQ-for-LLaMa](https://github.com/qwopqwop200/GPTQ-for-LLaMa).
 
+**This is an experimental new GPTQ which offers up to 8K context size**
+
+The increased context is currently only tested to work with [ExLlama](https://github.com/turboderp/exllama), via the latest release of [text-generation-webui](https://github.com/oobabooga/text-generation-webui).
+
+Please read carefully below to see how to use it.
+
+**NOTE**: Using the full 8K context will exceed 24GB VRAM.
+
 ## Repositories available
 
 * [4-bit GPTQ models for GPU inference](https://huggingface.co/TheBloke/WizardLM-33B-V1.0-Uncensored-SuperHOT-8KGPTQ)
@@ -34,73 +42,20 @@ It is the result of quantising to 4bit using [GPTQ-for-LLaMa](https://github.com
 Please make sure you're using the latest version of text-generation-webui
 
 1. Click the **Model tab**.
-2. Under **Download custom model or LoRA**, enter `TheBloke/WizardLM-33B-V1.0-Uncensored-SuperHOT-8KGPTQ`.
+2. Under **Download custom model or LoRA**, enter `TheBloke/WizardLM-33B-V1.0-Uncensored-SuperHOT-8K-GPTQ`.
 3. Click **Download**.
 4. The model will start downloading. Once it's finished it will say "Done"
-5. In the top left, click the refresh icon next to **Model**.
-6. In the **Model** dropdown, choose the model you just downloaded: `WizardLM-33B-V1.0-Uncensored-SuperHOT-8KGPTQ`
-7. The model will automatically load, and is now ready for use!
-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.
-  * 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`.
-9. Once you're ready, click the **Text Generation tab** and enter a prompt to get started!
-
-## How to use this GPTQ model from Python code
-
-First make sure you have [AutoGPTQ](https://github.com/PanQiWei/AutoGPTQ) installed:
-
-`pip install auto-gptq`
-
-Then try the following example code:
-
-```python
-from transformers import AutoTokenizer, pipeline, logging
-from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
-import argparse
-
-model_name_or_path = "TheBloke/WizardLM-33B-V1.0-Uncensored-SuperHOT-8KGPTQ"
-model_basename = "wizardlm-33b-v1.0-uncensored-superhot-8k-GPTQ-4bit--1g.act.order"
-
-use_triton = False
-
-tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
-
-model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
-        model_basename=model_basename,
-        use_safetensors=True,
-        trust_remote_code=False,
-        device="cuda:0",
-        use_triton=use_triton,
-        quantize_config=None)
-
-# Note: check the prompt template is correct for this model.
-prompt = "Tell me about AI"
-prompt_template=f'''USER: {prompt}
-ASSISTANT:'''
-
-print("\n\n*** Generate:")
-
-input_ids = tokenizer(prompt_template, return_tensors='pt').input_ids.cuda()
-output = model.generate(inputs=input_ids, temperature=0.7, max_new_tokens=512)
-print(tokenizer.decode(output[0]))
-
-# Inference can also be done using transformers' pipeline
-
-# Prevent printing spurious transformers error when using pipeline with AutoGPTQ
-logging.set_verbosity(logging.CRITICAL)
+5. Untick **Autoload the model**
+6. In the top left, click the refresh icon next to **Model**.
+7. In the **Model** dropdown, choose the model you just downloaded: `WizardLM-33B-V1.0-Uncensored-SuperHOT-8K-GPTQ`
+8. To use the increased context, set the **Loader** to **ExLlama**, set **max_seq_len** to 8192 or 4096, and set **compress_pos_emb** to **4** for 8192 context, or to **2** for 4096 context.
+9. Now click **Save Settings** followed by **Reload**
+10. The model will automatically load, and is now ready for use!
+11. Once you're ready, click the **Text Generation tab** and enter a prompt to get started!
 
-print("*** Pipeline:")
-pipe = pipeline(
-    "text-generation",
-    model=model,
-    tokenizer=tokenizer,
-    max_new_tokens=512,
-    temperature=0.7,
-    top_p=0.95,
-    repetition_penalty=1.15
-)
+## How to use this GPTQ model from Python code - TBC
 
-print(pipe(prompt_template)[0]['generated_text'])
-```
+Using this model with increased context from Python code is currently untested, so this section is removed for now.
 
 ## Provided files