diff --git a/src/diffusers/pipelines/unclip/pipeline_unclip.py b/src/diffusers/pipelines/unclip/pipeline_unclip.py index c9edcfd7aa78..53c4d201e35c 100644 --- a/src/diffusers/pipelines/unclip/pipeline_unclip.py +++ b/src/diffusers/pipelines/unclip/pipeline_unclip.py @@ -292,7 +292,7 @@ def __call__( # compute the previous noisy sample x_t -> x_t-1 decoder_latents = self.decoder_scheduler.step( - noise_pred, t, decoder_latents, prev_timestep=prev_timestep + noise_pred, t, decoder_latents, prev_timestep=prev_timestep, generator=generator ).prev_sample decoder_latents = decoder_latents.clamp(-1, 1) @@ -348,7 +348,7 @@ def __call__( # compute the previous noisy sample x_t -> x_t-1 super_res_latents = self.super_res_scheduler.step( - noise_pred, t, super_res_latents, prev_timestep=prev_timestep + noise_pred, t, super_res_latents, prev_timestep=prev_timestep, generator=generator ).prev_sample image = super_res_latents diff --git a/tests/pipelines/unclip/test_unclip.py b/tests/pipelines/unclip/test_unclip.py index ba49625511bc..a26c70d3749b 100644 --- a/tests/pipelines/unclip/test_unclip.py +++ b/tests/pipelines/unclip/test_unclip.py @@ -233,15 +233,15 @@ def test_unclip(self): expected_slice = np.array( [ - 0.0009, + 0.0011, + 0.0002, + 0.9962, + 0.9940, + 0.0002, 0.9997, 0.0003, - 0.9991, - 0.9967, - 0.0003, - 0.9997, - 0.0003, - 0.0004, + 0.9987, + 0.9989, ] ) @@ -261,7 +261,7 @@ def tearDown(self): def test_unclip_karlo(self): expected_image = load_numpy( "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main" - "/karlo_v1_alpha/horse.npy" + "/unclip/karlo_v1_alpha_horse.npy" ) pipeline = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha")