Unverified Commit 0ea53822 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[LongT5] Correct docs long t5 (#18669)

* add first generation tutorial

* [LongT5 Docs] Correct docs

* correct expected string

* remove incorrect file
parent 582c5371
...@@ -1966,13 +1966,12 @@ class LongT5ForConditionalGeneration(LongT5PreTrainedModel): ...@@ -1966,13 +1966,12 @@ class LongT5ForConditionalGeneration(LongT5PreTrainedModel):
... ) ... )
>>> # Let's try a very long input. >>> # Let's try a very long input.
>>> input_ids = tokenizer( >>> inputs = tokenizer(100 * "studies have shown that owning a dog is good for you ", return_tensors="pt")
... "summarize: " + 100 * "studies have shown that owning a dog is good for you ", return_tensors="pt" >>> input_ids = inputs.input_ids
... ).input_ids # Batch size 1
>>> outputs = model.generate(input_ids) >>> outputs = model.generate(input_ids)
>>> print(tokenizer.decode(outputs[0], skip_special_tokens=True)) >>> print(tokenizer.decode(outputs[0], skip_special_tokens=True))
abstractthe aim of this article is to summarize the studies have shown that owning a dog abstractthe aim of this article is to provide an overview of the literature on the role of dog
```""" ```"""
use_cache = use_cache if use_cache is not None else self.config.use_cache use_cache = use_cache if use_cache is not None else self.config.use_cache
return_dict = return_dict if return_dict is not None else self.config.use_return_dict return_dict = return_dict if return_dict is not None else self.config.use_return_dict
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment