Nicolas Iglesias commited on
Commit
ecf5055
·
1 Parent(s): 3e2d54d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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('### Respuesta:\n') + len('### Respuesta:\n')
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