electron-rare/exo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use half-open layer indexing in get_allow_patterns

Browse Source
This commit is contained in:
ciaranbor
2026-01-06 10:49:43 +00:00
parent f5eb5d0338
commit bcd69a3b01
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/exo/worker/download/huggingface_utils.py
+2 -2
View File
@@ -128,7 +128,7 @@ def get_allow_patterns(weight_map: dict[str, str], shard: ShardMetadata) -> list
layer_num = extract_layer_num(tensor_name_no_prefix)
if (
layer_num is not None
and shard.start_layer <= layer_num <= shard.end_layer
and shard.start_layer <= layer_num < shard.end_layer
):
shard_specific_patterns.add(filename)
@@ -144,7 +144,7 @@ def get_allow_patterns(weight_map: dict[str, str], shard: ShardMetadata) -> list
layer_num = extract_layer_num(tensor_name)
if (
layer_num is not None
and shard.start_layer <= layer_num <= shard.end_layer
and shard.start_layer <= layer_num < shard.end_layer
):
shard_specific_patterns.add(filename)
layer_independent_files = set(