try fix #1761 again (#1680)

lllyasviel · Dec 31, 2023 · 0c4f20a · 0c4f20a
1 parent 2521af3
commit 0c4f20a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/extras/resampler.py b/extras/resampler.py
@@ -108,8 +108,7 @@ def __init__(
             )
 
     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 @@ def forward(self, x):
             latents = ff(latents) + latents
 
         latents = self.proj_out(latents)
-        return self.norm_out(latents)
+        return self.norm_out(latents)
diff --git a/fooocus_version.py b/fooocus_version.py
@@ -1 +1 @@
-version = '2.1.858'
+version = '2.1.859'