Aleksandr Borzunov пре 2 година
родитељ
комит
1a34fd41b3
2 измењених фајлова са 2 додато и 4 уклоњено
  1. 0 1
      src/petals/__init__.py
  2. 2 3
      src/petals/server/throughput.py

+ 0 - 1
src/petals/__init__.py

@@ -2,5 +2,4 @@ from src.bloom import *
 from src.client import *
 from src.dht_utils import declare_active_modules, get_remote_module
 
-project_name = "bloomd"
 __version__ = "0.2"

+ 2 - 3
src/petals/server/throughput.py

@@ -11,7 +11,6 @@ from typing import Dict, Union
 import torch
 from hivemind.utils.logging import get_logger, use_hivemind_log_handler
 
-from src import project_name
 from src.bloom.block import BloomBlock
 from src.bloom.model import BloomConfig
 from src.bloom.ops import build_alibi_tensor
@@ -20,8 +19,8 @@ use_hivemind_log_handler("in_root_logger")
 logger = get_logger(__file__)
 
 
-DEFAULT_CACHE_PATH = Path(Path.home(), ".cache", project_name, "throughput.json")
-DEFAULT_LOCK_PATH = Path(tempfile.gettempdir(), project_name, "throughput.lock")
+DEFAULT_CACHE_PATH = Path(Path.home(), ".cache", "petals", "throughput.json")
+DEFAULT_LOCK_PATH = Path(tempfile.gettempdir(), "petals", "throughput.lock")
 
 SPEED_TEST_PATH = Path(Path(__file__).absolute().parents[2], "cli", "speed_test.py")