File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
src/compressed_tensors/utils Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 36
36
37
37
def match_named_modules (
38
38
model : torch .nn .Module ,
39
- targets : Iterable [str ] = tuple () ,
39
+ targets : Iterable [str ],
40
40
ignore : Iterable [str ] = tuple (),
41
41
warn_on_fail : bool = False ,
42
42
) -> Generator [Tuple [str , torch .nn .Module ]]:
@@ -69,7 +69,7 @@ def match_named_modules(
69
69
def match_named_parameters (
70
70
model : torch .nn .Module ,
71
71
targets : Iterable [str ],
72
- ignore : Iterable [str ],
72
+ ignore : Iterable [str ] = tuple () ,
73
73
warn_on_fail : bool = False ,
74
74
) -> Generator [Tuple [str , torch .nn .Module , torch .nn .Parameter ]]:
75
75
"""
@@ -103,7 +103,7 @@ def match_named_parameters(
103
103
def match_modules_set (
104
104
model : torch .nn .Module ,
105
105
targets : Iterable [str ],
106
- ignore : Iterable [str ],
106
+ ignore : Iterable [str ] = tuple () ,
107
107
) -> Generator [Iterable [torch .nn .Module ]]:
108
108
"""
109
109
Yields modules grouped with the same order and size as `targets`.
You can’t perform that action at this time.
0 commit comments