Give more memory in test_disk_offload (#25315)

fab1a0aa · Sylvain Gugger · GitHub · 67683095 · fab1a0aa
Unverified Commit fab1a0aa authored Aug 04, 2023 by Sylvain Gugger Committed by GitHub Aug 04, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_common.py tests/test_modeling_common.py +1 -1

No files found.
--- 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)