mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-01 19:49:03 +02:00
Merge release 1.28 (#1401)
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user