Update README.md
Browse files
README.md
CHANGED
@@ -42,7 +42,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
42 |
tokenizer = AutoTokenizer.from_pretrained("stabilityai/japanese-stablelm-base-gamma-7b")
|
43 |
model = AutoModelForCausalLM.from_pretrained(
|
44 |
"stabilityai/japanese-stablelm-base-gamma-7b",
|
45 |
-
trust_remote_code=True,
|
46 |
torch_dtype="auto",
|
47 |
)
|
48 |
model.cuda()
|
|
|
42 |
tokenizer = AutoTokenizer.from_pretrained("stabilityai/japanese-stablelm-base-gamma-7b")
|
43 |
model = AutoModelForCausalLM.from_pretrained(
|
44 |
"stabilityai/japanese-stablelm-base-gamma-7b",
|
|
|
45 |
torch_dtype="auto",
|
46 |
)
|
47 |
model.cuda()
|