Update README.md

README.md

@@ -39,7 +39,7 @@ input_ids=torch.tensor([tokenizer.encode(prefix_LM+lm_text)]).to(device)
 outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True)
 print(tokenizer.decode(outputs[0][1:]))
 
-# print result: с того, что он был в армии, служил в артиллерии.
+# print result: с того, что он был в плену у немцев.</s>
 
 #Prefix <SC1>
 lm_text='<SC1>Принялся Кутузов рассказывать свою историю . Началось с того, что он был в армии, служил в артиллерии.'
@@ -47,15 +47,15 @@ input_ids=torch.tensor([tokenizer.encode(prefix_LM+lm_text)]).to(device)
 outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True)
 print(tokenizer.decode(outputs[0][1:]))
 
-#print result: '<extra_id_0>  с самого начала'
+#print result: '<extra_id_0>, как он жил</s>'
 
-# Prefix <SC6>
-lm_text='<SC6>Принялся Кутузов рассказывать свою историю . Началось с того, что он был в армии, служил в артиллерии.'
+# Prefix <SC5>
+lm_text='<SC5>Принялся Кутузов рассказывать свою историю . Началось с того, что он был в армии, служил в артиллерии.'
 input_ids=torch.tensor([tokenizer.encode(prefix_LM+lm_text)]).to(device)
 outputs=model.generate(input_ids,eos_token_id=tokenizer.eos_token_id,early_stopping=True,max_length=100)
 print(tokenizer.decode(outputs[0][1:]))
 
-#print result:'<extra_id_0>.\n— Я родился в 1745 году, — начал он. — Отец мой был крестьянин, а мать — дочь священника. Отец мой был очень беден, и я с детства был предоставлен самому себе.\n— А как вы стали офицером? — спросил я.\n— Это длинная история'
+#print result: '<extra_id_0>, как он был в армии</s>'
 
 ```
 # Authors