Update README.md
Browse files
README.md
CHANGED
@@ -34,9 +34,7 @@ gen_kwargs = {'max_length': 256, 'top_p': 0.8, 'temperature':0.9, "num_beams": 1
|
|
34 |
prompt=['', 'MLFVVL', 'LDL', 'VTQA']
|
35 |
|
36 |
for idx, each in enumerate(prompt):
|
37 |
-
offset = 0
|
38 |
print(f"Begin generating idx: {idx} with prompt {each}")
|
39 |
-
output=''
|
40 |
output = model.chat(tokenizer, each)
|
41 |
print(f"\nEnd generation with length: {len(output.split())} - seqs: {output}\n")
|
42 |
```
|
|
|
34 |
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 |
```
|