Skip to content

Commit 43d931c

Browse files
committed
Merge branch 'main' into nightly
2 parents 0581621 + f3e9537 commit 43d931c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

‎unsloth/models/mapper.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1281,6 +1281,9 @@
12811281
for k in float8_values:
12821282
FLOAT_TO_FP8_BLOCK_MAPPER[k.lower()] = block
12831283
FLOAT_TO_FP8_ROW_MAPPER[k.lower()] = row
1284+
for k in float16_values:
1285+
FLOAT_TO_FP8_BLOCK_MAPPER[k.lower()] = block
1286+
FLOAT_TO_FP8_ROW_MAPPER[k.lower()] = row
12841287
values = float16_values
12851288
INT_TO_FLOAT_MAPPER[key] = values[0]
12861289

0 commit comments

Comments
 (0)