Create README.md

README.md

@@ -0,0 +1,125 @@
+---
+license: apache-2.0
+tags:
+- AWQ
+inference: false
+---
+
+# Falcon-40b-Instruct (4-bit 128g AWQ Quantized)
+[Falcon-40b-Instruct](https://huggingface.co/tiiuae/falcon-40b-instruct) is a 40B parameters causal decoder-only model built by [TII](https://www.tii.ae) based on [Falcon-40B](https://huggingface.co/tiiuae/falcon-7b) and finetuned on a mixture of chat/instruct datasets.
+
+This model is a 4-bit 128 group size AWQ quantized model. For more information about AWQ quantization, please click [here](https://github.com/mit-han-lab/llm-awq).
+
+## Model Date
+
+July 5, 2023
+
+## Model License
+
+Please refer to original Falcon model license ([link](https://huggingface.co/tiiuae/falcon-40b-instruct)).
+
+Please refer to the AWQ quantization license ([link](https://github.com/llm-awq/blob/main/LICENSE)).
+
+## CUDA Version
+
+This model was successfully tested on CUDA driver v530.30.02 and runtime v11.7 with Python v3.10.11. Please note that AWQ requires NVIDIA GPUs with compute capability of 80 or higher.
+
+## How to Use
+
+```bash
+git clone https://github.com/mit-han-lab/llm-awq \
+&& cd llm-awq \
+&& git checkout 71d8e68df78de6c0c817b029a568c064bf22132d \
+&& pip install -e . \
+&& cd awq/kernels \
+&& python setup.py install
+```
+
+```python
+import torch
+from awq.quantize.quantizer import real_quantize_model_weight
+from transformers import AutoModelForCausalLM, AutoConfig, AutoTokenizer
+from accelerate import init_empty_weights, load_checkpoint_and_dispatch
+from huggingface_hub import hf_hub_download
+
+model_name = "tiiuae/falcon-40b-instruct"
+
+# Config
+config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
+
+# Tokenizer
+tokenizer = AutoTokenizer.from_pretrained(config.tokenizer_name)
+
+# Model
+w_bit = 4
+q_config = {
+    "zero_point": True,
+    "q_group_size": 64,
+}
+
+load_quant = hf_hub_download('abhinavkulkarni/tiiuae-falcon-40b-instruct-w4-g128-awq', 'pytorch_model.bin')
+
+with init_empty_weights():
+    model = AutoModelForCausalLM.from_config(config=config, 
+                                                 torch_dtype=torch.float16, trust_remote_code=True)
+
+real_quantize_model_weight(model, w_bit=w_bit, q_config=q_config, init_only=True)
+
+model = load_checkpoint_and_dispatch(model, load_quant, device_map="balanced")
+
+# Inference
+prompt = f'''What is the difference between nuclear fusion and fission?
+###Response:'''
+
+input_ids = tokenizer(prompt, return_tensors='pt').input_ids.cuda()
+output = model.generate(
+    inputs=input_ids, 
+    temperature=0.7,
+    max_new_tokens=512,
+    top_p=0.15,
+    top_k=0,
+    repetition_penalty=1.1,
+    eos_token_id=tokenizer.eos_token_id
+)
+print(tokenizer.decode(output[0], skip_special_tokens=True))
+```
+
+## Evaluation
+
+This evaluation was done using [LM-Eval](https://github.com/EleutherAI/lm-evaluation-harness).
+
+[Falcon-40b-Instruct](https://huggingface.co/tiiuae/falcon-40b-instruct)
+
+
+[Falcon-40b-Instruct (4-bit 128-group AWQ)](https://huggingface.co/abhinavkulkarni/tiiuae-falcon-40b-instruct-w4-g128-awq)
+
+|  Task  |Version|    Metric     |Value |   |Stderr|
+|--------|------:|---------------|-----:|---|------|
+|wikitext|      1|word_perplexity|8.9237|   |      |
+|        |       |byte_perplexity|1.5058|   |      |
+|        |       |bits_per_byte  |0.5905|   |      |
+
+
+## Acknowledgements
+
+*Paper coming soon* 😊. In the meanwhile, you can use the following information to cite: 
+```
+@article{falcon40b,
+  title={{Falcon-40B}: an open large language model with state-of-the-art performance},
+  author={Almazrouei, Ebtesam and Alobeidli, Hamza and Alshamsi, Abdulaziz and Cappelli, Alessandro and Cojocaru, Ruxandra and Debbah, Merouane and Goffinet, Etienne and Heslow, Daniel and Launay, Julien and Malartic, Quentin and Noune, Badreddine and Pannier, Baptiste and Penedo, Guilherme},
+  year={2023}
+}
+```
+
+
+The model was quantized with AWQ technique. If you find AWQ useful or relevant to your research, please kindly cite the paper:
+
+```
+@article{lin2023awq,
+  title={AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration},
+  author={Lin, Ji and Tang, Jiaming and Tang, Haotian and Yang, Shang and Dang, Xingyu and Han, Song},
+  journal={arXiv},
+  year={2023}
+}
+```
+