AMajor/grok-1
1
0
Fork 0
mirror of https://github.com/xai-org/grok-1.git synced 2024-11-22 19:49:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Compare commits

base: AMajor:a105de1c162a250a7db6271c3e10a11979892fc6
Branches Tags
AMajor:main
AMajor:download-instruction
...
compare: AMajor:0b0576a0a857cbdb4e3be851b39005e8ebdb3f97
Branches Tags
AMajor:main
AMajor:download-instruction

2 Commits
a105de1c16 ... 0b0576a0a8

Author SHA1 Message Date
Madhav Kumar
0b0576a0a8
Merge 104affb51a into 7050ed204b 2024-03-21 06:48:52 +05:30
Madhav Kumar
104affb51a
Set maxsize to 32MB in lru_cache() for reasonable memory usage 2024-03-21 06:48:22 +05:30
1 changed files with 1 additions and 1 deletions
Show Stats Expand all files Collapse all files

2
checkpoint.py
View File

@ -141,7 +141,7 @@ def path_tuple_to_string(path: tuple) -> str:
For get_load_path_str(),
introducing a simple caching mechanism to avoid recomputing regex matches for paths that have already been processed.
"""
@lru_cache(maxsize=None)
@lru_cache(maxsize=32) # Set maxsize to 32MB
def get_load_path_str_cached(
init_path_str: str,
load_rename_rules: Optional[list[tuple[str, str]]] = None,