Kukedlc commited on
Commit
456cecd
1 Parent(s): 506491e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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'