[Bugfix] Race condition in MultiprocExecutor when concurent access to Scheduler (#1448)
Signed-off-by: knlnguyen1802 <knlnguyen1802@gmail.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
1 parent
03c01e0415
commit
5361353be5
4 files changed
+504
-45
No files matched your search
@@ -143,7 +143,10 @@ class DiffusionWorker:
|
||||
self.rank,
|
||||
process_memory / GiB_bytes,
|
||||
)
|
||||
assert self.model_runner.pipeline is not None
|
||||
|
||||
# When load_format is "dummy", pipeline will init with custom pipeline later
|
||||
if load_format != "dummy":
|
||||
assert self.model_runner.pipeline is not None
|
||||
|
||||
def init_lora_manager(self) -> None:
|
||||
"""Initialize the LoRA manager for this worker."""
|
||||
|
||||
Reference in new issue
Block a user