Upload anytext.py

anytext.py

@@ -868,9 +868,9 @@ class AuxiliaryLatentModule(nn.Module):
             new_string += char + " " * nSpace
         return new_string[:-nSpace]
 
-    def to(self, device):
-        self.device = device
-        self.vae = self.vae.to(device)
+    def to(self, *args, **kwargs):
+        self.vae = self.vae.to(*args, **kwargs)
+        self.device = self.vae.device
         return self