AquilaX-AI
/

DB-Summarizer

Model card Files Files and versions Community

Mr-Vicky-01 commited on 13 days ago

Commit

f4dd327

·

verified ·

1 Parent(s): 2c167ff

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -35,7 +35,7 @@ encodeds = tokenizer(summ_inp, return_tensors="pt",truncation=True).input_ids.to
 model.to(device)
 inputs = inputs.to(device)
 text_streamer = TextStreamer(tokenizer, skip_prompt = True)
-response = finetuned_model.generate(
         input_ids=encodeds,
         streamer=text_streamer,
         max_new_tokens=512,

 model.to(device)
 inputs = inputs.to(device)
 text_streamer = TextStreamer(tokenizer, skip_prompt = True)
+response = model.generate(
         input_ids=encodeds,
         streamer=text_streamer,
         max_new_tokens=512,