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-02-11 06:14:27 +00:00
commit 02e94ce17a
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
unsloth/models/loader.py
View file

@ -293,9 +293,11 @@ class FastLanguageModel(FastLlamaModel):
# AMD GPT-OSS routing:
# - Radeon can often use prequantized bnb-4bit checkpoints.
# - Instinct/MI (warp=64) often cannot, so fallback to BF16.
if is_hip() and (
"gpt-oss" in model_name.lower() or "gpt_oss" in model_name.lower()
) and not use_exact_model_name:
if (
is_hip()
and ("gpt-oss" in model_name.lower() or "gpt_oss" in model_name.lower())
and not use_exact_model_name
):
gpt_oss_prequant_suffix = model_name.lower().endswith(
("-unsloth-bnb-4bit", "-bnb-4bit")
)
@ -901,9 +903,11 @@ class FastModel(FastBaseModel):
# AMD GPT-OSS routing:
# - Radeon can often use prequantized bnb-4bit checkpoints.
# - Instinct/MI (warp=64) often cannot, so fallback to BF16.
if is_hip() and (
"gpt-oss" in model_name.lower() or "gpt_oss" in model_name.lower()
) and not use_exact_model_name:
if (
is_hip()
and ("gpt-oss" in model_name.lower() or "gpt_oss" in model_name.lower())
and not use_exact_model_name
):
gpt_oss_prequant_suffix = model_name.lower().endswith(
("-unsloth-bnb-4bit", "-bnb-4bit")
)