.

vince62s · Nov 11, 2023 · d357503 · d357503
1 parent 9019b12
commit d357503
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onmt/models/model.py b/onmt/models/model.py
@@ -120,7 +120,7 @@ def load_state_dict(
             offset = 0
         buf_list = []
         for buf_name, buf in self.named_buffers():
-                buf_list.append(buf_name)
+            buf_list.append(buf_name)
         for name, module in self.named_modules():
             named_buf_and_param = list(module.named_buffers()) + list(
                 module.named_parameters()