Unverified Commit c3ecf2d9 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix doctest CI (#26324)



fix doc CI
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 06ee91ae
...@@ -790,11 +790,11 @@ class PersimmonForCausalLM(PersimmonPreTrainedModel): ...@@ -790,11 +790,11 @@ class PersimmonForCausalLM(PersimmonPreTrainedModel):
```python ```python
>>> from transformers import AutoTokenizer, PersimmonForCausalLM >>> from transformers import AutoTokenizer, PersimmonForCausalLM
>>> model = PersimmonForCausalLM.from_pretrained("ArthurZ/persimmon-8b-base").cuda() >>> model = PersimmonForCausalLM.from_pretrained("ArthurZ/persimmon-8b-base")
>>> tokenizer = AutoTokenizer.from_pretrained("ArthurZ/persimmon-8b-base") >>> tokenizer = AutoTokenizer.from_pretrained("ArthurZ/persimmon-8b-base")
>>> prompt = "human: Hey, what should I eat for dinner?" >>> prompt = "human: Hey, what should I eat for dinner?"
>>> inputs = tokenizer(prompt, return_tensors="pt").cuda() >>> inputs = tokenizer(prompt, return_tensors="pt")
>>> # Generate >>> # Generate
>>> generate_ids = model.generate(inputs.input_ids, max_length=30) >>> generate_ids = model.generate(inputs.input_ids, max_length=30)
......
...@@ -710,6 +710,7 @@ src/transformers/models/pegasus_x/modeling_pegasus_x.py ...@@ -710,6 +710,7 @@ src/transformers/models/pegasus_x/modeling_pegasus_x.py
src/transformers/models/perceiver/configuration_perceiver.py src/transformers/models/perceiver/configuration_perceiver.py
src/transformers/models/perceiver/convert_perceiver_haiku_to_pytorch.py src/transformers/models/perceiver/convert_perceiver_haiku_to_pytorch.py
src/transformers/models/persimmon/convert_persimmon_weights_to_hf.py src/transformers/models/persimmon/convert_persimmon_weights_to_hf.py
src/transformers/models/persimmon/modeling_persimmon.py
src/transformers/models/pix2struct/configuration_pix2struct.py src/transformers/models/pix2struct/configuration_pix2struct.py
src/transformers/models/pix2struct/convert_pix2struct_original_pytorch_to_hf.py src/transformers/models/pix2struct/convert_pix2struct_original_pytorch_to_hf.py
src/transformers/models/pix2struct/image_processing_pix2struct.py src/transformers/models/pix2struct/image_processing_pix2struct.py
......
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