Nuclear Ai
commited on
Commit
•
6497484
1
Parent(s):
2f33490
Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ We Used : [NuclearAi/Nuke-Python-Verse](https://huggingface.co/datasets/NuclearA
|
|
30 |
|
31 |
# Here is the code to run it in 4-bit Quantization method !
|
32 |
|
33 |
-
```
|
34 |
```python
|
35 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextStreamer
|
36 |
import torch
|
@@ -93,4 +93,4 @@ while True:
|
|
93 |
# The TextStreamer already prints the response token by token, so we just print a newline
|
94 |
print() # Ensure to move to the next line after the response is printed
|
95 |
```
|
96 |
-
```
|
|
|
30 |
|
31 |
# Here is the code to run it in 4-bit Quantization method !
|
32 |
|
33 |
+
```python
|
34 |
```python
|
35 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextStreamer
|
36 |
import torch
|
|
|
93 |
# The TextStreamer already prints the response token by token, so we just print a newline
|
94 |
print() # Ensure to move to the next line after the response is printed
|
95 |
```
|
96 |
+
```
|