Selaa lähdekoodia

Rename src.server.{load_balancing => block_selection}

Aleksandr Borzunov 3 vuotta sitten
vanhempi
commit
abcfb17a10
2 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  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__)