Nicolas Iglesias
commited on
Commit
·
ecf5055
1
Parent(s):
3e2d54d
Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ with torch.no_grad():
|
|
64 |
|
65 |
s = generation_output.sequences[0]
|
66 |
output = tokenizer.decode(s)
|
67 |
-
start_txt = output.find('
|
68 |
end_txt = output.find("<|endoftext|>", start_txt)
|
69 |
answer = output[start_txt:end_txt]
|
70 |
|
|
|
64 |
|
65 |
s = generation_output.sequences[0]
|
66 |
output = tokenizer.decode(s)
|
67 |
+
start_txt = output.find('[/INST]') + len('[/INST]')
|
68 |
end_txt = output.find("<|endoftext|>", start_txt)
|
69 |
answer = output[start_txt:end_txt]
|
70 |
|