[Feature] Support minicpmv v2.6 (#2785)
Co-authored-by: Chayenne <zhaochen20@outlook.com> Co-authored-by: yizhang2077 <1109276519@qq.com>
This commit is contained in:
@@ -237,7 +237,7 @@ class ModelRunner:
|
||||
set_custom_all_reduce(not self.server_args.disable_custom_all_reduce)
|
||||
|
||||
if not self.is_draft_worker:
|
||||
# Only initilzie the distributed environment on the target model worker.
|
||||
# Only initialize the distributed environment on the target model worker.
|
||||
init_distributed_environment(
|
||||
backend=backend,
|
||||
world_size=self.tp_size,
|
||||
|
||||
Reference in New Issue
Block a user