Ty_Tech/heretic
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: fix formatting and naming

Browse Source
This commit is contained in:
Philipp Emanuel Weidmann
2026-03-06 13:18:08 +05:30
parent 5e3c04c802
commit ec0367226d
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/heretic/model.py
+12 -7
View File
@@ -166,13 +166,18 @@ class Model:
# because hybrid models like Qwen3.5 MoE have modules with different names # because hybrid models like Qwen3.5 MoE have modules with different names
# across layers (e.g. "o_proj" on attention layers, "out_proj" on linear attention layers). # across layers (e.g. "o_proj" on attention layers, "out_proj" on linear attention layers).
target_modules_set: set[str] = set() target_modules_set: set[str] = set()
layers = self.get_layers()
for layer_index, layer in enumerate(layers): for layer_index, layer in enumerate(self.get_layers()):
module_id_to_leaf_name = {id(m): name.split(".")[-1] for name, m in layer.named_modules()} module_id_to_leaf_name = {
for modules_list in self.get_layer_modules(layer_index).values(): id(module): module_name.split(".")[-1]
for mod in modules_list: for module_name, module in layer.named_modules()
if id(mod) in module_id_to_leaf_name: }
target_modules_set.add(module_id_to_leaf_name[id(mod)])
for modules in self.get_layer_modules(layer_index).values():
for module in modules:
if id(module) in module_id_to_leaf_name:
target_modules_set.add(module_id_to_leaf_name[id(module)])
target_modules = list(target_modules_set) target_modules = list(target_modules_set)
if self.settings.row_normalization != RowNormalization.FULL: if self.settings.row_normalization != RowNormalization.FULL: