Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name,
|
|
31 |
torch_dtype=torch.bfloat16,
|
32 |
)
|
33 |
|
34 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
35 |
|
36 |
model.config.use_cache = True
|
37 |
def stream(user_prompt):
|
|
|
31 |
torch_dtype=torch.bfloat16,
|
32 |
)
|
33 |
|
34 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name,add_bos_token=True,trust_remote_code=True)
|
35 |
|
36 |
model.config.use_cache = True
|
37 |
def stream(user_prompt):
|