瀏覽代碼

Rename src.server.{load_balancing => block_selection}

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

+ 0 - 0
src/server/load_balancing.py → src/server/block_selection.py


+ 1 - 1
src/server/server.py

@@ -16,9 +16,9 @@ from src.bloom.from_pretrained import DTYPE_MAP, load_pretrained_block
 from src.data_structures import CHAIN_DELIMITER, UID_DELIMITER, ServerState
 from src.dht_utils import get_remote_module_infos
 from src.server.backend import TransformerBackend
+from src.server.block_selection import choose_best_blocks
 from src.server.cache import MemoryCache
 from src.server.handler import TransformerConnectionHandler
-from src.server.load_balancing import choose_best_blocks
 
 use_hivemind_log_handler("in_root_logger")
 logger = get_logger(__file__)