Update README.md
Browse files
README.md
CHANGED
@@ -14,11 +14,11 @@ import torch
|
|
14 |
|
15 |
|
16 |
model = AutoModelForCausalLM.from_pretrained(
|
17 |
-
'TwT-6/
|
18 |
attn_implementation="flash_attention_2",
|
19 |
trust_remote_code=True, torch_dtype=torch.bfloat16, device_map="auto").eval()
|
20 |
|
21 |
-
tokenizer = AutoTokenizer.from_pretrained('TwT-6/
|
22 |
|
23 |
inputs = '你好'
|
24 |
|
|
|
14 |
|
15 |
|
16 |
model = AutoModelForCausalLM.from_pretrained(
|
17 |
+
'TwT-6/cr-model',
|
18 |
attn_implementation="flash_attention_2",
|
19 |
trust_remote_code=True, torch_dtype=torch.bfloat16, device_map="auto").eval()
|
20 |
|
21 |
+
tokenizer = AutoTokenizer.from_pretrained('TwT-6/cr-model', trust_remote_code=True)
|
22 |
|
23 |
inputs = '你好'
|
24 |
|