Update README.md
Browse files
README.md
CHANGED
@@ -40,8 +40,8 @@ def generate_prompt(text):
|
|
40 |
### Response:"""
|
41 |
|
42 |
|
43 |
-
tokenizer = LlamaTokenizer.from_pretrained('minlik/chinese-alpaca-
|
44 |
-
model = LlamaForCausalLM.from_pretrained('minlik/chinese-alpaca-
|
45 |
model.eval()
|
46 |
|
47 |
text = '第一个登上月球的人是谁?'
|
|
|
40 |
### Response:"""
|
41 |
|
42 |
|
43 |
+
tokenizer = LlamaTokenizer.from_pretrained('minlik/chinese-alpaca-33b-merged')
|
44 |
+
model = LlamaForCausalLM.from_pretrained('minlik/chinese-alpaca-33b-merged').half().to('cuda')
|
45 |
model.eval()
|
46 |
|
47 |
text = '第一个登上月球的人是谁?'
|