VideoMAE doctest - use valid dummy pixel values (#22022)

Use valid dummy pixel values

VideoMAE doctest - use valid dummy pixel values (#22022)
Use valid dummy pixel values
4130e703 · amyeroberts · GitHub · c1f85598 · 4130e703
Unverified Commit 4130e703 authored Mar 08, 2023 by amyeroberts Committed by GitHub Mar 08, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/videomae/modeling_videomae.py src/transformers/models/videomae/modeling_videomae.py +1 -1

No files found.
--- a/src/transformers/models/videomae/modeling_videomae.py
+++ b/src/transformers/models/videomae/modeling_videomae.py
@@ -790,7 +790,7 @@ class VideoMAEForPreTraining(VideoMAEPreTrainedModel):
        >>> import torch

        >>> num_frames = 16
-        >>> video = list(np.random.randn(16, 3, 224, 224))
+        >>> video = list(np.random.randint(0, 256, (num_frames, 3, 224, 224)))

        >>> image_processor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base")
        >>> model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")