import torch
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained(
    "google/pegasus-cnn_dailymail"
)
model = AutoModelForSeq2SeqLM.from_pretrained(
    "google/pegasus-cnn_dailymail",
    dtype=torch.float16,
    device_map="auto",
    attn_implementation="sdpa"
)

input_text = """Plants are remarkable organisms that produce their own food using a method called photosynthesis.
This process involves converting sunlight, carbon dioxide, and water into glucose, which provides energy for growth.
Plants play a crucial role in sustaining life on Earth by generating oxygen and serving as the foundation of most ecosystems."""
input_ids = tokenizer(input_text, return_tensors="pt").to(model.device)

output = model.generate(**input_ids, cache_implementation="static")
print(tokenizer.decode(output[0], skip_special_tokens=True))

---

Plants are remarkable organisms that produce their own food using a method called photosynthesis.<n>This process involves converting sunlight, carbon dioxide, and water into glucose, which provides energy for growth.

---

And want seek help Jen first gavepist brought well magical am high place go her three Brit Meadows building – event masterre – variety source me well only nourished recruitmentA am many high place go her three Brit Meadows building masterre – variety source me well only nourished recruitmentA am many high place go her three Brit Meadows buildingpre – variety source me well only nourished recruitmentA am many high place go her three Brit Meadows buildingpre – variety source me well only healthy am many high place go her three Brit Meadows buildingpre – variety source me well only healthy am many high place go her three Brit Meadows buildingpre –

---

Plants are remarkable organisms that produce their own food using a method called photosynthesis.<n>This process involves converting sunlight, carbon dioxide, and water into glucose, which provides energy for growth.