瀏覽代碼

Fix AssertionError on rebalancing (#370)

Alexander Borzunov 2 年之前
父節點
當前提交
a6fdfc0556
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/petals/server/server.py

+ 1 - 1
src/petals/server/server.py

@@ -418,7 +418,7 @@ class ModuleContainer(threading.Thread):
         module_uids = [f"{dht_prefix}{UID_DELIMITER}{block_index}" for block_index in block_indices]
         memory_cache = MemoryCache(attn_cache_bytes, alloc_timeout)
 
-        assert server_info.state == ServerState.JOINING
+        server_info.state = ServerState.JOINING
         dht_announcer = ModuleAnnouncerThread(
             module_uids,
             dht,