Compare commits

...

2 Commits

Author SHA1 Message Date
Sylvain Gugger
5e3b19a805 Patch release: v4.27.3
Some checks failed
Release - Conda / build_and_package (push) Has been cancelled
2023-03-23 14:04:40 -04:00
Sylvain Gugger
62d9baa53c Enforce max_memory for device_map strategies (#22311)
Enforce  for device_map strategies
2023-03-23 14:04:10 -04:00
3 changed files with 3 additions and 2 deletions

View File

@@ -418,7 +418,7 @@ install_requires = [
setup(
name="transformers",
version="4.27.2", # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
version="4.27.3", # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
author="The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/transformers/graphs/contributors)",
author_email="transformers@huggingface.co",
description="State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow",

View File

@@ -18,7 +18,7 @@
# to defer the actual importing for when the objects are requested. This way `import transformers` provides the names
# in the namespace without actually importing anything (and especially none of the backends).
__version__ = "4.27.2"
__version__ = "4.27.3"
from typing import TYPE_CHECKING

View File

@@ -2576,6 +2576,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
model,
dtype=torch_dtype,
low_zero=(device_map == "balanced_low_0"),
max_memory=max_memory,
**kwargs,
)
kwargs["max_memory"] = max_memory