npvinHnivqn
commited on
Commit
•
770e490
1
Parent(s):
41cc240
Update README.md
Browse files
README.md
CHANGED
@@ -42,9 +42,9 @@ This format is also available through the tokenizer's `apply_chat_template` meth
|
|
42 |
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModelForSeq2SeqLM
|
43 |
import torch
|
44 |
|
45 |
-
tokenizer = AutoTokenizer.from_pretrained('
|
46 |
model = AutoModelForCausalLM.from_pretrained(
|
47 |
-
'
|
48 |
trust_remote_code=True,
|
49 |
device_map="auto"
|
50 |
)
|
|
|
42 |
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModelForSeq2SeqLM
|
43 |
import torch
|
44 |
|
45 |
+
tokenizer = AutoTokenizer.from_pretrained('npvinHnivqn/stablelm_zephyr_3b_with_cross_attn')
|
46 |
model = AutoModelForCausalLM.from_pretrained(
|
47 |
+
'npvinHnivqn/stablelm_zephyr_3b_with_cross_attn',
|
48 |
trust_remote_code=True,
|
49 |
device_map="auto"
|
50 |
)
|