SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Use PyAV instead of Decord in examples (#21572)

Browse Source 
* Use PyAV instead of Decord

* Get frame indices

* Fix number of frames

* Update src/transformers/models/videomae/image_processing_videomae.py

* Fix up

* Fix copies

* Update timesformer doctests

* Update docstrings
This commit is contained in:
amyeroberts
2023-03-02 12:30:38 +00:00
committed by GitHub
parent c256bc6d10
commit 3412f5979d
8 changed files with 224 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@@ -98,6 +98,7 @@ if stale_egg_info.exists():
_deps = [
"Pillow",
"accelerate>=0.10.0",
"av==9.2.0", # Latest version of PyAV (10.0.0) has issues with audio stream.
"beautifulsoup4",
"black~=23.1",
"codecarbon==1.2.0",
@@ -289,7 +290,7 @@ extras["timm"] = deps_list("timm")
extras["torch-vision"] = deps_list("torchvision") + extras["vision"]
extras["natten"] = deps_list("natten")
extras["codecarbon"] = deps_list("codecarbon")
extras["video"] = deps_list("decord")
extras["video"] = deps_list("decord", "av")
extras["sentencepiece"] = deps_list("sentencepiece", "protobuf")
extras["testing"] = (