fix formatting

basedformer/optimizer.py

@@ -41,12 +41,15 @@ class BasedOptimizer:
             
         if optimizer == "adamw":
             self.optimizer = optim.AdamW(parameters, lr=0, weight_decay=self.weight_decay, betas=(self.beta1, self.beta2), eps=self.eps)
+
         elif optimizer == "adamw8bit":
             import bitsandbytes as bnb
             self.optimizer = bnb.optim.Adam8bit(parameters, lr=0, weight_decay=self.weight_decay, betas=(self.beta1, self.beta2), eps=self.eps)
+
         elif optimizer == "adafactor":
             try:
                 from transformers.optimization import Adafactor
+
             except ImportError:
                 raise ImportError("Please install transformers for Adafactor")
 
@@ -55,6 +58,7 @@ class BasedOptimizer:
     def step(self, scaler=None):
         if scaler:
             scaler.step(self.optimizer)
+            
         else:
             self.optimizer.step()