diff --git a/lmms_eval/tasks/FALCONBench/utils.py b/lmms_eval/tasks/FALCONBench/utils.py index aad1329c3..5568791b3 100644 --- a/lmms_eval/tasks/FALCONBench/utils.py +++ b/lmms_eval/tasks/FALCONBench/utils.py @@ -25,7 +25,7 @@ import requests import torch import yaml -from decord import cpu +from decord import VideoReader, cpu from loguru import logger as eval_logger from PIL import Image diff --git a/lmms_eval/tasks/longvideobench/utils.py b/lmms_eval/tasks/longvideobench/utils.py index 8a58fc465..fc95ae383 100644 --- a/lmms_eval/tasks/longvideobench/utils.py +++ b/lmms_eval/tasks/longvideobench/utils.py @@ -7,7 +7,7 @@ import torch import yaml -from decord import cpu +from decord import VideoReader, cpu from loguru import logger as eval_logger from PIL import Image