groupuser/stablelm-2-1_6b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(README): remove trust_remote_code requirement from tokenizer snippet

Browse Source
This commit is contained in:
Jonathan Tow 2024-03-01 07:35:30 +00:00 committed by system
parent a7a1fb8a83
commit db5a120c4d
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -30,7 +30,7 @@ Get started generating text with `Stable LM 2 1.6B` by using the following code
```python ```python
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-1_6b", trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-1_6b")
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
"stabilityai/stablelm-2-1_6b", "stabilityai/stablelm-2-1_6b",
torch_dtype="auto", torch_dtype="auto",
@ -54,7 +54,7 @@ print(tokenizer.decode(tokens[0], skip_special_tokens=True))
```python ```python
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-1_6b", trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-1_6b")
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
"stabilityai/stablelm-2-1_6b", "stabilityai/stablelm-2-1_6b",
torch_dtype="auto", torch_dtype="auto",