accelerate

22bf677c - Allow the torch device to be set with an env var (#1009)

Allow the torch device to be set with an env var (#1009) * Allow the torch device to be set with an env var Signed-off-by: Antoni Baum <antoni.baum@protonmail.com> * Fix Signed-off-by: Antoni Baum <antoni.baum@protonmail.com> * Refactor Signed-off-by: Antoni Baum <antoni.baum@protonmail.com> * Use self.device Signed-off-by: Antoni Baum <antoni.baum@protonmail.com> * Refactor Signed-off-by: Antoni Baum <antoni.baum@protonmail.com> * Add test * Add test * Fix test * Tweak comment * Fix test Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>