add to imports

stanford-crfm · Jan 23, 2025 · 603fdd1 · 603fdd1
1 parent 3a5d4ff
commit 603fdd1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/levanter/optim/__init__.py b/src/levanter/optim/__init__.py
@@ -1,4 +1,4 @@
-from .config import AdamConfig, OptimizerConfig
+from .config import AdamConfig, OptimizerConfig, SGDConfig
 from .sophia import (  # SophiaGConfig,; SophiaGObjective,
     ScaleBySophiaState,
     SophiaHConfig,