ai-forever commited on
Commit
336d15f
1 Parent(s): d86532a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,7 +50,7 @@ input_ids=torch.tensor([tokenizer.encode(lm_text)]).to(device)
50
  outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True)
51
  print(tokenizer.decode(outputs[0][1:]))
52
 
53
- #print result: '<extra_id_0> с самого начала</s>'
54
 
55
  # Prefix <SC5>
56
  lm_text='<SC5>Принялся Кутузов рассказывать свою историю <extra_id_0>. Началось с того, что он был в армии, служил в артиллерии.'
@@ -58,7 +58,7 @@ input_ids=torch.tensor([tokenizer.encode(lm_text)]).to(device)
58
  outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True)
59
  tokenizer.decode(outputs[0][1:])
60
 
61
- #print result: '<extra_id_0>, и она оказалась очень длинной</s>'
62
 
63
  ```
64
  # Authors
 
50
  outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True)
51
  print(tokenizer.decode(outputs[0][1:]))
52
 
53
+ #print result: '<extra_id_0>, как он воевал</s>'
54
 
55
  # Prefix <SC5>
56
  lm_text='<SC5>Принялся Кутузов рассказывать свою историю <extra_id_0>. Началось с того, что он был в армии, служил в артиллерии.'
 
58
  outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True)
59
  tokenizer.decode(outputs[0][1:])
60
 
61
+ #print result: '<extra_id_0>, как он стал генералом</s>'
62
 
63
  ```
64
  # Authors