From fab1a0aa82e032eee3816d1c767ec9e95659fb8f Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Fri, 4 Aug 2023 14:10:31 +0200 Subject: [PATCH] Give more memory in test_disk_offload (#25315) --- tests/test_modeling_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 0cdc94fc8e4..fe2e2bc8195 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -2521,7 +2521,7 @@ class ModelTesterMixin: base_output = model(**inputs_dict_class) model_size = compute_module_sizes(model)[""] - max_size = int(self.model_split_percents[0] * model_size) + max_size = int(self.model_split_percents[1] * model_size) with tempfile.TemporaryDirectory() as tmp_dir: model.cpu().save_pretrained(tmp_dir)