We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 336cc5b commit 23eb2d5Copy full SHA for 23eb2d5
grail/infrastructure/checkpoint_consumer.py
@@ -46,8 +46,8 @@
46
from grail.shared.safetensors_utils import load_model_state_dict
47
48
from ..shared.constants import (
49
+ BASE_CHECKPOINT_RETENTION_LIMIT,
50
CHECKPOINT_MILESTONE_INTERVAL,
- CHECKPOINT_RETENTION_LIMIT,
51
DELTA_BASE_INTERVAL,
52
GRAIL_CHECKPOINT_MOD10,
53
WINDOW_LENGTH,
@@ -115,7 +115,7 @@ def __init__(
115
*,
116
cache_root: Path,
117
credentials: Any | None,
118
- keep_limit: int = CHECKPOINT_RETENTION_LIMIT,
+ keep_limit: int = BASE_CHECKPOINT_RETENTION_LIMIT,
119
) -> None:
120
self.cache_root = cache_root.expanduser().resolve()
121
self.cache_root.mkdir(parents=True, exist_ok=True)
0 commit comments