Drop MI300X arch suffix in banner name
This commit is contained in:
parent
ec93910e0a
commit
bb6962e107
1 changed files with 1 additions and 1 deletions
2
unsloth/models/_utils.py
Normal file → Executable file
2
unsloth/models/_utils.py
Normal file → Executable 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 (
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue