dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix doctest CI (#26324) 

fix doc CI

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2023-09-22 08:58:30 +02:00 committed by GitHub
parent 06ee91aebc
commit c3ecf2d95d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/persimmon/modeling_persimmon.py
View File

@ -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)

1
utils/not_doctested.txt
View File

@ -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