zR
commited on
Commit
·
0cbeda4
1
Parent(s):
b531660
test
Browse files- README.md +1 -2
- README_zh.md +1 -2
README.md
CHANGED
|
@@ -122,8 +122,7 @@ inputs = tokenizer.apply_chat_template(
|
|
| 122 |
return_tensors='pt',
|
| 123 |
add_generation_prompt=True,
|
| 124 |
return_dict=True,
|
| 125 |
-
|
| 126 |
-
)
|
| 127 |
|
| 128 |
input_len = inputs['input_ids'].shape[1]
|
| 129 |
generate_kwargs = {
|
|
|
|
| 122 |
return_tensors='pt',
|
| 123 |
add_generation_prompt=True,
|
| 124 |
return_dict=True,
|
| 125 |
+
).to(model.device)
|
|
|
|
| 126 |
|
| 127 |
input_len = inputs['input_ids'].shape[1]
|
| 128 |
generate_kwargs = {
|
README_zh.md
CHANGED
|
@@ -94,8 +94,7 @@ inputs = tokenizer.apply_chat_template(
|
|
| 94 |
return_tensors='pt',
|
| 95 |
add_generation_prompt=True,
|
| 96 |
return_dict=True,
|
| 97 |
-
|
| 98 |
-
)
|
| 99 |
|
| 100 |
input_len = inputs['input_ids'].shape[1]
|
| 101 |
generate_kwargs = {
|
|
|
|
| 94 |
return_tensors='pt',
|
| 95 |
add_generation_prompt=True,
|
| 96 |
return_dict=True,
|
| 97 |
+
).to(model.device)
|
|
|
|
| 98 |
|
| 99 |
input_len = inputs['input_ids'].shape[1]
|
| 100 |
generate_kwargs = {
|