From 78e89ce8b5610a71005e14a3b7df20d1fd7321d1 Mon Sep 17 00:00:00 2001 From: disperaller Date: Wed, 22 May 2024 13:55:05 +0800 Subject: [PATCH] change max_task_size to dynamic value --- opencompass/partitioners/size.py | 1 + 1 file changed, 1 insertion(+) diff --git a/opencompass/partitioners/size.py b/opencompass/partitioners/size.py index 56659ca0..3efba82a 100644 --- a/opencompass/partitioners/size.py +++ b/opencompass/partitioners/size.py @@ -1,6 +1,7 @@ import copy import math import os.path as osp +import torch from fnmatch import fnmatch from typing import Dict, List, Optional, Tuple, Union