update other platforms' Dockerfile
This commit is contained in:
@@ -358,7 +358,7 @@ class NPUWorker(WorkerBase):
|
||||
def try_reload_vram(self) -> tuple[bool, bool]:
|
||||
allocator = CaMemAllocator.get_instance()
|
||||
return allocator.try_reload_vram(tags=None)
|
||||
|
||||
|
||||
def vnpu_unlock_gpu(self) -> None:
|
||||
allocator = CaMemAllocator.get_instance()
|
||||
allocator.vnpu_unlock_gpu()
|
||||
|
||||
Reference in New Issue
Block a user