File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
vllm/distributed/device_communicators Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -108,8 +108,14 @@ def __init__(self,
108
108
# created by the process. The following patch is a workaround.
109
109
with patch ("multiprocessing.resource_tracker.register" ,
110
110
lambda * args , ** kwargs : None ):
111
- self .shared_memory = shared_memory .SharedMemory (name = name )
112
- assert self .shared_memory .size == self .total_bytes_of_buffer
111
+ try :
112
+ self .shared_memory = shared_memory .SharedMemory (name = name )
113
+ assert self .shared_memory .size == self .total_bytes_of_buffer # noqa
114
+ except FileNotFoundError :
115
+ # we might deserialize the object in a different node
116
+ # in this case, this object is not used,
117
+ # and we should suppress the error
118
+ pass
113
119
114
120
def __reduce__ (self ):
115
121
return (
You can’t perform that action at this time.
0 commit comments