Skip to content

Commit 07df691

Browse files
fix(postprocess): correct _merge_buffer_entries call convention (#47)
_merge_buffer_entries is a module-level function, not a method on ModelHistory. Fixes AttributeError when processing recurrent models with duplicate buffer entries. Based on gilmoright's contribution in PR #56. Co-authored-by: gilmoright <artem.dahaka@gmail.com>
1 parent 3738449 commit 07df691

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchlens/postprocess.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1221,7 +1221,7 @@ def _fix_buffer_layers(self):
12211221
and (unique_buffer.tensor_contents is not None)
12221222
and (torch.equal(buffer.tensor_contents, unique_buffer.tensor_contents))
12231223
):
1224-
self._merge_buffer_entries(unique_buffer, buffer)
1224+
_merge_buffer_entries(self, unique_buffer, buffer)
12251225
break
12261226
unique_buffers.append(buffer_label)
12271227

0 commit comments

Comments
 (0)