Skip to content

Commit 6ecb2bc

Browse files
committed
Fix formatting
Signed-off-by: Po-Han Huang <[email protected]>
1 parent b33f11b commit 6ecb2bc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/model_executor/models/llama4.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -375,7 +375,8 @@ def is_fp4_weight(weight):
375375
new_loaded_weight = new_loaded_weight[shard_idx]
376376

377377
# Only transpose for non-FP4 weights
378-
# FP4 weights are already in the correct format and shouldn't be transposed here.
378+
# FP4 weights are already in the correct format and
379+
# shouldn't be transposed here.
379380
if not is_fp4_weight(new_loaded_weight):
380381
new_loaded_weight = new_loaded_weight.transpose(-1, -2)
381382

0 commit comments

Comments
 (0)