ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-04-16 21:55:14 +00:00
commit e264d42413
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
unsloth/models/loader.py
View file

@ -1384,10 +1384,14 @@ class FastModel(FastBaseModel):
global FORCE_BFLOAT16
for disable_name in FORCE_BFLOAT16:
if (
disable_name.lower()
== model_type_arch.lower().replace("-", "").replace("_", "")
or disable_name.lower() in model_types_all
) and dtype == torch.float16 and SUPPORTS_BFLOAT16:
(
disable_name.lower()
== model_type_arch.lower().replace("-", "").replace("_", "")
or disable_name.lower() in model_types_all
)
and dtype == torch.float16
and SUPPORTS_BFLOAT16
):
logger.warning_once(
f"Unsloth: {model_type_arch} does not support float16 training. "
f"Switching to bfloat16."