From 4130e7036793ab2da08c1f3992affa1e56349ef8 Mon Sep 17 00:00:00 2001 From: amyeroberts <22614925+amyeroberts@users.noreply.github.com> Date: Wed, 8 Mar 2023 11:54:42 +0000 Subject: [PATCH] VideoMAE doctest - use valid dummy pixel values (#22022) Use valid dummy pixel values --- src/transformers/models/videomae/modeling_videomae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/videomae/modeling_videomae.py b/src/transformers/models/videomae/modeling_videomae.py index 0d1f12b8730..2ec9c01e6f0 100644 --- 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")