abhinavkulkarni
/

mosaicml-mpt-7b-instruct-w4-g128-awq

Text Generation

text-generation-inference

Model card Files Files and versions Community

Abhinav Kulkarni commited on Jul 20, 2023

Commit

dc8a9d0

•

1 Parent(s): 38afe4a

Updated README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ git clone https://github.com/abhinavkulkarni/llm-awq \
 ```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 snapshot_download

 ```python
 import torch
 from awq.quantize.quantizer import real_quantize_model_weight
+from transformers import AutoModelForCausalLM, AutoConfig, AutoTokenizer, TextStreamer
 from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 from huggingface_hub import snapshot_download