Skip to content

Commit

Permalink
ignore lint
Browse files Browse the repository at this point in the history
  • Loading branch information
RunDevelopment committed Mar 25, 2024
1 parent 4598054 commit 83db003
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ def load(self, state_dict: StateDict) -> ImageModelDescriptor[MIRNet2]:
n_feat = state_dict["conv_in.weight"].shape[0]
bias = "conv_in.bias" in state_dict

n_MRB = get_seq_len(state_dict, "body.0.body") - 1
n_MRB = get_seq_len(state_dict, "body.0.body") - 1 # noqa: N806
chan_factor = (
state_dict["body.0.body.0.dau_mid.body.0.weight"].shape[0] / n_feat
)
Expand Down

0 comments on commit 83db003

Please sign in to comment.