Skip to content

GitLab

Unverified Commit fab1a0aa authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Give more memory in test_disk_offload (#25315)

parent 67683095
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_common.py
View file @ fab1a0aa
...@@ -2521,7 +2521,7 @@ class ModelTesterMixin: ...@@ -2521,7 +2521,7 @@ class ModelTesterMixin:
base_output = model(**inputs_dict_class) base_output = model(**inputs_dict_class)
model_size = compute_module_sizes(model)[""] 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: with tempfile.TemporaryDirectory() as tmp_dir:
model.cpu().save_pretrained(tmp_dir) model.cpu().save_pretrained(tmp_dir)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment