Update README.md

README.md

@@ -35,8 +35,8 @@ import torch
 num_frames = 16
 video = list(np.random.randn(16, 3, 224, 224))
 
-feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base")
-model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")
+feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-short")
+model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base-short")
 
 pixel_values = feature_extractor(video, return_tensors="pt").pixel_values
 
@@ -48,7 +48,7 @@ outputs = model(pixel_values, bool_masked_pos=bool_masked_pos)
 loss = outputs.loss
 ```
 
-For more code examples, we refer to the [documentation](https://huggingface.co/transformers/model_doc/vit.html#).
+For more code examples, we refer to the [documentation](https://huggingface.co/transformers/main/model_doc/videomae.html#).
 
 ## Training data