Fix bug where SetFitHead not moved to non-cuda devices on init (#518)

Co-authored-by: Tom Aarsen <[email protected]>

Author: ajmssc

src/setfit/modeling.py

@@ -91,7 +91,7 @@ def __init__(
         self.temperature = temperature
         self.eps = eps
         self.bias = bias
-        self._device = device or "cuda" if torch.cuda.is_available() else "cpu"
+        self._device = device or ("cuda" if torch.cuda.is_available() else "cpu")
         self.multitarget = multitarget
 
         self.to(self._device)