Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
dtlzhuangz committed Apr 22, 2024
1 parent c8fb808 commit 2306f84
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/transformers/integrations/eetq.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,8 +40,8 @@ def _replace_with_eetq_linear(
"""
if current_key_name is None:
current_key_name = []
for name, module in model.named_children():

for name, module in model.named_children():
current_key_name.append(name)

if (isinstance(module, nn.Linear)) and name not in modules_to_not_convert:
Expand Down

0 comments on commit 2306f84

Please sign in to comment.