From 603fdd186972be134a6bcfa5090bebaf0815d6d3 Mon Sep 17 00:00:00 2001 From: Suhas Kotha Date: Thu, 23 Jan 2025 15:12:39 -0800 Subject: [PATCH] add to imports --- src/levanter/optim/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/levanter/optim/__init__.py b/src/levanter/optim/__init__.py index 7dec2ebb4..632e25f02 100644 --- 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,