fix ovis2 device_map (#3496)

Author: Jintao-Huang

swift/trainers/trainers.py

@@ -208,6 +208,7 @@ def compute_loss(self, model, inputs, return_outputs=False, num_items_in_batch=N
 
         if labels is None:
             labels = inputs['labels']
+            outputs.loss = outputs.loss.to(labels.device)
             # fix https://github.com/huggingface/transformers/issues/34263
             if num_items_in_batch is not None:
                 outputs.loss = outputs.loss * (labels[:, 1:] != -100).sum() / num_items_in_batch