We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 281cd67 commit a4698b9Copy full SHA for a4698b9
lightllm/utils/dist_utils.py
@@ -55,7 +55,7 @@ def get_environ(environ_name):
55
56
def init_vision_distributed_env(kvargs):
57
tp_world_size = kvargs["vit_tp"]
58
- dp_size = kvargs.get("dp_size", 1)
+ dp_size = 1
59
tp_rank_id = kvargs["tp_rank_id"]
60
set_dp_size(dp_size)
61
set_dp_world_size(tp_world_size)
0 commit comments