Merge release 1.28 (#1401)

This commit is contained in:
jinghanhu
2025-08-04 17:00:14 +08:00
committed by GitHub
parent 7ed173be8b
commit b537efebed
2 changed files with 2 additions and 2 deletions

View File

@@ -277,7 +277,7 @@ class LLMImageBuilder(Builder):
if not args.vllm_version:
args.vllm_version = '0.8.5.post1'
if not args.lmdeploy_version:
args.lmdeploy_version = '0.7.2.post1'
args.lmdeploy_version = '0.9.1'
if not args.autogptq_version:
args.autogptq_version = '0.7.1'
if not args.flashattn_version:

View File

@@ -277,7 +277,7 @@ def custom_group_norm(input,
"""
b, c, h, w = input.shape
channel_in_group = c // num_groups
input_reshaped = input.reshape(1, b * num_groups, channel_in_group, h, w)
input_reshaped = input.reshape(b * num_groups, channel_in_group, h, w)
out = F.batch_norm(
input_reshaped,