From eb1e0f907079a942831751863054672ef9eb6622 Mon Sep 17 00:00:00 2001 From: Pu Fanyi Date: Thu, 26 Dec 2024 23:58:05 +0800 Subject: [PATCH] Update oryx.py --- lmms_eval/models/oryx.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/lmms_eval/models/oryx.py b/lmms_eval/models/oryx.py index 5d4dfcf20..9e1a860e3 100644 --- a/lmms_eval/models/oryx.py +++ b/lmms_eval/models/oryx.py @@ -22,6 +22,14 @@ import os import sys +os.environ["LOWRES_RESIZE"] = "384x32" +os.environ["VIDEO_RESIZE"] = "0x64" +os.environ["HIGHRES_BASE"] = "0x32" +os.environ["MAXRES"] = "1536" +os.environ["MINRES"] = "0" +os.environ["VIDEO_MAXRES"] = "480" +os.environ["VIDEO_MINRES"] = "288" + try: from oryx.constants import ( DEFAULT_IM_END_TOKEN, @@ -471,6 +479,8 @@ def generate_until(self, requests) -> List[str]: res.append(outputs) pbar.update(1) except Exception as e: + import traceback + traceback.print_exc() eval_logger.info(f"{e}") eval_logger.info(f"Video {visuals} generate failed, check the source") video_path = "\n".join(visuals)