Compare commits

...

3 Commits

Author SHA1 Message Date
b1c9710894 the actual change? 2025-10-03 14:59:20 +00:00
eff732fa46 Merge branch 'main' into default-auto 2025-10-03 14:43:29 +00:00
b6062082af default to "auto" dtype 2024-11-25 15:45:12 +01:00

View File

@ -4469,7 +4469,7 @@ class PreTrainedModel(nn.Module, EmbeddingAccessMixin, ModuleUtilsMixin, PushToH
use_auth_token = kwargs.pop("use_auth_token", None)
from_pipeline = kwargs.pop("_from_pipeline", None)
from_auto_class = kwargs.pop("_from_auto", False)
dtype = kwargs.pop("dtype", None)
dtype = kwargs.pop("dtype", "auto")
torch_dtype = kwargs.pop("torch_dtype", None) # kept for BC
device_map = kwargs.pop("device_map", None)
max_memory = kwargs.pop("max_memory", None)