abhinavkulkarni
/

psmathur-orca_mini_v2_7b-w4-g128-awq

Text Generation

text-generation-inference

Model card Files Files and versions Community

Abhinav Kulkarni commited on Jul 20, 2023

Commit

62e2cab

•

1 Parent(s): 75ebf2d

Updated README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,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