diff --git a/src/nanotron/serialize/__init__.py b/src/nanotron/serialize/__init__.py
index ae6ef264..7fc7b0a9 100644
--- a/src/nanotron/serialize/__init__.py
+++ b/src/nanotron/serialize/__init__.py
@@ -2,3 +2,4 @@
 from nanotron.serialize.main import *
 from nanotron.serialize.optimizer import *
 from nanotron.serialize.random import *
+from nanotron.serialize.weights import *