more fixing

basedformer/optimizer.py

@@ -58,7 +58,7 @@ class BasedOptimizer:
     def step(self, scaler=None):
         if scaler:
             scaler.step(self.optimizer)
-            
+
         else:
             self.optimizer.step()
             
@@ -94,5 +94,6 @@ class BasedOptimizer:
     def load(cls, path):
         with open(path, 'rb') as f:
             based_optimizer = pickle.load(f)
+            
         based_optimizer.optimizer.load_state_dict(torch.load(path))
         return based_optimizer
\ No newline at end of file