update: SFWrapper

Author: kozistr

pytorch_optimizer/optimizer/schedulefree.py

@@ -594,9 +594,6 @@ def step(self, closure: CLOSURE = None) -> LOSS:
 
                 state = self.state[p]
 
-                if 'z' not in state:
-                    state['z'] = torch.clone(p)
-
                 z = state['z']
 
                 self.apply_weight_decay(