We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 912bc56 + 1c4eec2 commit d340aa1Copy full SHA for d340aa1
1 file changed
src/maxtext/integration/tunix/weight_mapping/gpt_oss.py
@@ -80,7 +80,7 @@ def to_hf_mapping(
80
target_indices = range(start, start + layers_per_block)
81
82
regex_indices = "|".join(map(str, target_indices))
83
- layer_regex = f"layers\.({regex_indices})"
+ layer_regex = rf"layers\.({regex_indices})"
84
85
# --- 3. Block Mappings (Standard) ---
86
mapping.update(
0 commit comments