Abhinav Kulkarni
commited on
Commit
•
6dcea68
1
Parent(s):
1dbe901
Updated README
Browse files
README.md
CHANGED
@@ -40,6 +40,7 @@ git clone https://github.com/mit-han-lab/llm-awq \
|
|
40 |
```
|
41 |
|
42 |
```python
|
|
|
43 |
import torch
|
44 |
from awq.quantize.quantizer import real_quantize_model_weight
|
45 |
from transformers import AutoModelForCausalLM, AutoConfig, AutoTokenizer, TextStreamer
|
|
|
40 |
```
|
41 |
|
42 |
```python
|
43 |
+
import time
|
44 |
import torch
|
45 |
from awq.quantize.quantizer import real_quantize_model_weight
|
46 |
from transformers import AutoModelForCausalLM, AutoConfig, AutoTokenizer, TextStreamer
|