Abhinav Kulkarni commited on
Commit
0d484bb
1 Parent(s): 26df441

Updated README

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ git clone https://github.com/abhinavkulkarni/llm-awq \
45
  ```python
46
  import torch
47
  from awq.quantize.quantizer import real_quantize_model_weight
48
- from transformers import AutoModelForCausalLM, AutoConfig, AutoTokenizer
49
  from accelerate import init_empty_weights, load_checkpoint_and_dispatch
50
  from huggingface_hub import snapshot_download
51
 
 
45
  ```python
46
  import torch
47
  from awq.quantize.quantizer import real_quantize_model_weight
48
+ from transformers import AutoModelForCausalLM, AutoConfig, AutoTokenizer, TextStreamer
49
  from accelerate import init_empty_weights, load_checkpoint_and_dispatch
50
  from huggingface_hub import snapshot_download
51