[Misc] [CI] Fix CI failure caught after merge (#4126)

This commit is contained in:
Cade Daniel
2024-04-16 17:56:01 -07:00
committed by GitHub
parent e95cd87959
commit d150e4f89f

View File

@ -68,6 +68,7 @@ class GPUExecutor(ExecutorBase):
scheduler_config=self.scheduler_config,
device_config=self.device_config,
cache_config=self.cache_config,
load_config=self.load_config,
local_rank=0,
rank=0,
distributed_init_method=distributed_init_method,
@ -82,6 +83,7 @@ class GPUExecutor(ExecutorBase):
scheduler_config=self.scheduler_config,
device_config=self.device_config,
cache_config=self.cache_config,
load_config=self.load_config,
local_rank=0,
rank=0,
distributed_init_method=distributed_init_method,