[Bug]: schedule-free needs version upgrade

[dxqbYD]

What happened?

Discord user report:

Traceback (most recent call last):
  File "C:\Users\rolan\AppData\Local\Programs\Python\Python310\lib\threading.py", line 1016, in _bootstrap_inner
    self.run()
  File "C:\Users\rolan\AppData\Local\Programs\Python\Python310\lib\threading.py", line 953, in run
    self._target(*self._args, **self._kwargs)
  File "C:\Users\rolan\Desktop\OneTrainer-master\modules\ui\TrainUI.py", line 566, in training_thread_function
    trainer.end()
  File "C:\Users\rolan\Desktop\OneTrainer-master\modules\trainer\GenericTrainer.py", line 754, in end
    self.backup(self.model.train_progress)
  File "C:\Users\rolan\Desktop\OneTrainer-master\modules\trainer\GenericTrainer.py", line 424, in backup
    self.model.optimizer.eval()
  File "C:\Users\rolan\Desktop\OneTrainer-master\venv\lib\site-packages\schedulefree\adamwschedulefree.py", line 85, in eval
    p.data.lerp(end=state['z'], weight=1-1/beta1)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

What did you expect would happen?

schedule-free seems to need an upgrade, because they have changed this line since 1.3:
facebookresearch/schedule_free@5225d9b

latest version is 1.4

Relevant log output

Output of pip freeze

No response