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'
|