Abhinav Kulkarni commited on
Commit
7d178c8
1 Parent(s): 1b396be

Updated README

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -23,7 +23,7 @@ Please refer to the AWQ quantization license ([link](https://github.com/llm-awq/
23
 
24
  ## CUDA Version
25
 
26
- 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.
27
 
28
  For Docker users, the `nvcr.io/nvidia/pytorch:23.06-py3` image is runtime v12.1 but otherwise the same as the configuration above and has also been verified to work.
29
 
@@ -84,7 +84,7 @@ output = model.generate(
84
  repetition_penalty=1.1,
85
  eos_token_id=tokenizer.eos_token_id
86
  )
87
- print(tokenizer.decode(output[0], skip_special_tokens=True))
88
  ```
89
 
90
  ## Evaluation
 
23
 
24
  ## CUDA Version
25
 
26
+ 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 `8.0` or higher.
27
 
28
  For Docker users, the `nvcr.io/nvidia/pytorch:23.06-py3` image is runtime v12.1 but otherwise the same as the configuration above and has also been verified to work.
29
 
 
84
  repetition_penalty=1.1,
85
  eos_token_id=tokenizer.eos_token_id
86
  )
87
+ # print(tokenizer.decode(output[0], skip_special_tokens=True))
88
  ```
89
 
90
  ## Evaluation