load

bghira · Aug 29, 2024 · 1d92148 · 1d92148
1 parent e86100f
commit 1d92148
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/discord_tron_client/classes/image_manipulation/pipeline_runners/base_runner.py b/discord_tron_client/classes/image_manipulation/pipeline_runners/base_runner.py
@@ -150,7 +150,7 @@ def load_adapter(
             else:
                 # lycoris_wrapper.to(self.pipeline.transformer.device)
                 lycoris_wrapper.apply_to()
-                self.pipeline.to(self.pipeline_manager.device)
+                lycoris_wrapper.to(device=self.pipeline_manager.device, dtype=self.pipeline_manager.torch_dtype)
         self.loaded_adapters[clean_adapter_name] = {
             "adapter_type": adapter_type,
             "adapter_path": adapter_path,