Forturne
/

KPMG-NARVIS-summarization

text2text-generation

Model card Files Files and versions

Forturne commited on Feb 13, 2023

Commit

ab5b12a

·

1 Parent(s): f323473

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -7,6 +7,7 @@ from transformers import BartForConditionalGeneration
 tokenizer = PreTrainedTokenizerFast.from_pretrained('Forturne/KPMG-NARVIS-summarization')
 model = BartForConditionalGeneration.from_pretrained('Forturne/KPMG-NARVIS-summarization')
 device="cuda:0"
 model.to(device)

 tokenizer = PreTrainedTokenizerFast.from_pretrained('Forturne/KPMG-NARVIS-summarization')
 model = BartForConditionalGeneration.from_pretrained('Forturne/KPMG-NARVIS-summarization')
+model.eval()
 device="cuda:0"
 model.to(device)