.

huggingface · Nov 22, 2024 · f00a380 · f00a380
1 parent ca0ffa6
commit f00a380
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 *