From 7e5cacff1ac3132a3ac21d95af306ea2c09bf276 Mon Sep 17 00:00:00 2001
From: lrast <https://github.com/lrast>
Date: Fri, 18 Apr 2025 16:06:34 -0400
Subject: [PATCH] Uses pytorch AdamW instead of transformers, which is
 deprecated

---
 examples/optimizer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/optimizer.py b/examples/optimizer.py
index 8e5ef065..396062bc 100644
--- a/examples/optimizer.py
+++ b/examples/optimizer.py
@@ -1,5 +1,5 @@
-from torch.optim import SGD, Adam
-from transformers import AdamW
+from torch.optim import SGD, Adam, AdamW
+
 
 def initialize_optimizer(config, model):
     # initialize optimizers