diff --git a/extras/resampler.py b/extras/resampler.py
index 4521c8c3..539f309d 100644
--- a/extras/resampler.py
+++ b/extras/resampler.py
@@ -108,8 +108,7 @@ class Resampler(nn.Module):
             )
 
     def forward(self, x):
-        
-        latents = self.latents.repeat(x.size(0), 1, 1)
+        latents = self.latents.repeat(x.size(0), 1, 1).to(x)
         
         x = self.proj_in(x)
         
@@ -118,4 +117,4 @@ class Resampler(nn.Module):
             latents = ff(latents) + latents
             
         latents = self.proj_out(latents)
-        return self.norm_out(latents)
\ No newline at end of file
+        return self.norm_out(latents)
diff --git a/fooocus_version.py b/fooocus_version.py
index 66847e88..8cde204d 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1 +1 @@
-version = '2.1.858'
+version = '2.1.859'