Drop MI300X arch suffix in banner name

This commit is contained in:
Daniel Hanchen 2026-02-14 06:52:20 +00:00
commit bb6962e107

2
unsloth/models/_utils.py Normal file → Executable file
View file

@ -176,7 +176,7 @@ def resolve_hip_gpu_stats_name(gpu_stats):
if arch_name:
# gfx942 maps to MI300X on current ROCm naming.
if arch_name.lower().startswith("gfx942"):
return f"AMD Instinct MI300X ({arch_name}). "
return "AMD Instinct MI300X. "
return f"AMD GPU ({arch_name}). "
return "AMD GPU Device. "
from unsloth_zoo.temporary_patches import (