From 0c4f20a0d261d4fe045aae6e5a0fcad185ba6804 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 31 Dec 2023 04:38:31 -0800 Subject: [PATCH] try fix #1761 again (#1680) --- extras/resampler.py | 5 ++--- fooocus_version.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) 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'