try fix #1761 again (#1680)

2023-12-31 04:38:31 -08:00 · 2023-12-31 04:38:31 -08:00 · 0c4f20a0d2
parent 2521af308b
commit 0c4f20a0d2
2 changed files with 3 additions and 4 deletions
--- 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)
+        return self.norm_out(latents)
--- a/fooocus_version.py
+++ b/fooocus_version.py
@ -1 +1 @@
-version = '2.1.858'
+version = '2.1.859'