IlyaGusev
/

rugpt3medium_sum_gazeta

text-generation

text-generation-inference

Model card Files Files and versions Community

IlyaGusev commited on Nov 26, 2021

Commit

4445e35

•

1 Parent(s): 44dc49d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ input_ids = torch.LongTensor([input_ids])
 output_ids = model.generate(
     input_ids=input_ids,
     no_repeat_ngram_size=4
-)[0]
 summary = tokenizer.decode(output_ids[0], skip_special_tokens=False)
 summary = summary.split(tokenizer.sep_token)[1]

 output_ids = model.generate(
     input_ids=input_ids,
     no_repeat_ngram_size=4
+)
 summary = tokenizer.decode(output_ids[0], skip_special_tokens=False)
 summary = summary.split(tokenizer.sep_token)[1]