Apply suggestions from code review

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: AlexeyKozhevin

batchflow/models/torch/base.py

@@ -633,7 +633,7 @@ def _parse_devices(self):
                             if device not in self.devices[:i]]
             self.device = self.devices[0]
 
-        if self.device.type != 'cuda':
+        if self.device.type == 'cpu':
             #TODO: maybe, we should add warning
             self.amp = False
         if torch.cuda.is_available():