Update README.md
Browse files
README.md
CHANGED
|
@@ -59,8 +59,12 @@ dtype: bfloat16
|
|
| 59 |
# Requirements
|
| 60 |
!pip install -qU transformers accelerate bitsandbytes
|
| 61 |
|
| 62 |
-
# Imports
|
| 63 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
|
|
|
|
|
|
|
|
|
|
|
|
| 64 |
|
| 65 |
# Model & Tokenizer
|
| 66 |
MODEL_NAME = 'Kukedlc/NeuralTopBench-7B-ties'
|
|
|
|
| 59 |
# Requirements
|
| 60 |
!pip install -qU transformers accelerate bitsandbytes
|
| 61 |
|
| 62 |
+
# Imports & settings
|
| 63 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
| 64 |
+
import warnings
|
| 65 |
+
import os
|
| 66 |
+
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
| 67 |
+
warnings.filterwarnings('ignore')
|
| 68 |
|
| 69 |
# Model & Tokenizer
|
| 70 |
MODEL_NAME = 'Kukedlc/NeuralTopBench-7B-ties'
|