From 2fb135df03ac97f3c5dc50822b7f9b2ed30d17aa Mon Sep 17 00:00:00 2001 From: aszc-dev Date: Fri, 17 May 2024 13:55:56 +0200 Subject: [PATCH] Fix set_timestamps for new LCMScheduler implementation --- coreml_suite/lcm/converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/coreml_suite/lcm/converter.py b/coreml_suite/lcm/converter.py index e79e761..6d3c3fc 100644 --- a/coreml_suite/lcm/converter.py +++ b/coreml_suite/lcm/converter.py @@ -63,7 +63,7 @@ def get_encoder_hidden_states_shape(unet_config, batch_size): def get_scheduler(): scheduler = LCMScheduler.from_pretrained(MODEL_VERSION, subfolder="scheduler") - scheduler.set_timesteps(50, get_torch_device(), 50, 1.0) + scheduler.set_timesteps(50, get_torch_device(), 50) return scheduler