fix torch_dtype & cpu_text_encoder

iopaint/model/helper/cpu_text_encoder.py

@@ -8,6 +8,7 @@ class CPUTextEncoderWrapper(PreTrainedModel):
     def __init__(self, text_encoder, torch_dtype):
         super().__init__(text_encoder.config)
         self.config = text_encoder.config
+        self._device = text_encoder.device
         # cpu not support float16
         self.text_encoder = text_encoder.to(torch.device("cpu"), non_blocking=True)
         self.text_encoder = self.text_encoder.to(torch.float32, non_blocking=True)
@@ -30,3 +31,11 @@ class CPUTextEncoderWrapper(PreTrainedModel):
     @property
     def dtype(self):
         return self.torch_dtype
+
+    @property
+    def device(self) -> torch.device:
+        """
+        `torch.device`: The device on which the module is (assuming that all the module parameters are on the same
+        device).
+        """
+        return self._device