riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Give more memory in test_disk_offload (#25315)

Browse Source
This commit is contained in:
Sylvain Gugger 2023-08-04 14:10:31 +02:00 committed by GitHub
parent 67683095a6
commit fab1a0aa82
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_common.py
View File

@ -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)