Update README.md

README.md

@@ -37,7 +37,7 @@ model.to(device)
 
 #Prefix <LM>
 lm_text='<LM>Принялся Кутузов рассказывать свою историю как он сюда попал. Началось'
-input_ids=torch.tensor([tokenizer.encode(prefix_LM+lm_text)]).to(device)
+input_ids=torch.tensor([tokenizer.encode(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:]))
 
@@ -45,7 +45,7 @@ print(tokenizer.decode(outputs[0][1:]))
 
 #Prefix <SC1>
 lm_text='<SC1>Принялся Кутузов рассказывать свою историю <extra_id_0>. Началось с того, что он был в армии, служил в артиллерии.'
-input_ids=torch.tensor([tokenizer.encode(prefix_LM+lm_text)]).to(device)
+input_ids=torch.tensor([tokenizer.encode(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:]))
 
@@ -53,7 +53,7 @@ print(tokenizer.decode(outputs[0][1:]))
 
 # Prefix <SC5>
 lm_text='<SC5>Принялся Кутузов рассказывать свою историю <extra_id_0>. Началось с того, что он был в армии, служил в артиллерии.'
-input_ids=torch.tensor([tokenizer.encode(prefix_LM+lm_text)]).to(device)
+input_ids=torch.tensor([tokenizer.encode(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:]))