Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ prompt=['', 'MLFVVL', 'LDL', 'VTQA']
|
|
| 35 |
|
| 36 |
for idx, each in enumerate(prompt):
|
| 37 |
print(f"Begin generating idx: {idx} with prompt {each}")
|
| 38 |
-
output = model.chat(tokenizer, each)
|
| 39 |
print(f"\nEnd generation with length: {len(output.split())} - seqs: {output}\n")
|
| 40 |
```
|
| 41 |
|
|
|
|
| 35 |
|
| 36 |
for idx, each in enumerate(prompt):
|
| 37 |
print(f"Begin generating idx: {idx} with prompt {each}")
|
| 38 |
+
output = model.chat(tokenizer, each, **gen_kwargs)
|
| 39 |
print(f"\nEnd generation with length: {len(output.split())} - seqs: {output}\n")
|
| 40 |
```
|
| 41 |
|