spolivin
/

bart-arxiv-lora

Model card Files Files and versions

spolivin commited on Mar 9, 2025

Commit

c79868f

·

verified ·

1 Parent(s): 489cf75

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -70,6 +70,6 @@ inputs = tokenizer(
 lora_model.to(inputs.input_ids.device)
 # Generating summarized version
-summary_ids = lora_model.generate(**inputs, max_length=max_length, num_beams=4, early_stopping=True)
 summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 ```

 lora_model.to(inputs.input_ids.device)
 # Generating summarized version
+summary_ids = lora_model.generate(**inputs, max_length=250, num_beams=4, early_stopping=True)
 summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 ```