Skip to content

Commit a4698b9

Browse files
huochaitiantangliuliang1
and
liuliang1
authored
Fix vision dp_size = 1 (#785)
Co-authored-by: liuliang1 <[email protected]>
1 parent 281cd67 commit a4698b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightllm/utils/dist_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def get_environ(environ_name):
5555

5656
def init_vision_distributed_env(kvargs):
5757
tp_world_size = kvargs["vit_tp"]
58-
dp_size = kvargs.get("dp_size", 1)
58+
dp_size = 1
5959
tp_rank_id = kvargs["tp_rank_id"]
6060
set_dp_size(dp_size)
6161
set_dp_world_size(tp_world_size)

0 commit comments

Comments
 (0)