Fix the deprecation of the usage Accelerator, add code for new way

ManifoldRG / NEKO

In Progress Implementation of GATO style Generalist Multimodal model capable of image, text, RL and Robotics tasks

https://discord.gg/brsPnzNd8h

GNU General Public License v3.0

43 stars 10 forks source link

Fix the deprecation of the usage Accelerator, add code for new way #85

Closed henryj18 closed 4 months ago

henryj18 commented 6 months ago

Change the original train.py to cope with the following Huggingface PR https://github.com/huggingface/accelerate/pull/2441 The original code will throw warning and stop running without the fix

eihli commented 4 months ago

Oh. Just noticed we need to bump the version of accelerate in env.yml.

(neko) ➜  accelerate git:(main) git tag --contains $(git log -G DataLoaderConfiguration --reverse --format='%H' | head -n1) | cat
v0.28.0
v0.29.0
v0.29.1
v0.29.2
v0.29.3
v0.30.0

-    - accelerate==0.20.3
+    - accelerate==0.30.0