riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix ImageGPT doctest (#24353)

Browse Source 
Fix doctest
This commit is contained in:
amyeroberts 2023-06-19 15:23:29 +01:00 committed by GitHub
parent a4de24f691
commit 7e71eb2ef7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/imagegpt/modeling_imagegpt.py
View File

@ -980,7 +980,7 @@ class ImageGPTForCausalImageModeling(ImageGPTPreTrainedModel):
>>> image_processor = AutoImageProcessor.from_pretrained("openai/imagegpt-small") >>> image_processor = AutoImageProcessor.from_pretrained("openai/imagegpt-small")
>>> model = ImageGPTForCausalImageModeling.from_pretrained("openai/imagegpt-small") >>> model = ImageGPTForCausalImageModeling.from_pretrained("openai/imagegpt-small")
>>> device = torch.device("cuda" if torch.cuda.is_available() else "cpu") >>> device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
>>> model.to(device) >>> model.to(device) # doctest: +IGNORE_RESULT
>>> # unconditional generation of 8 images >>> # unconditional generation of 8 images
>>> batch_size = 4 >>> batch_size = 4
@ -1000,7 +1000,7 @@ class ImageGPTForCausalImageModeling(ImageGPTPreTrainedModel):
... ] # convert color cluster tokens back to pixels ... ] # convert color cluster tokens back to pixels
>>> f, axes = plt.subplots(1, batch_size, dpi=300) >>> f, axes = plt.subplots(1, batch_size, dpi=300)
>>> for img, ax in zip(samples_img, axes): >>> for img, ax in zip(samples_img, axes): # doctest: +IGNORE_RESULT
... ax.axis("off") ... ax.axis("off")
... ax.imshow(img) ... ax.imshow(img)
```""" ```"""