pass device to processors Annotator

2026-03-21 08:08:13 +00:00 · 2024-07-01 17:37:25 +08:00
parent 996515c7ca
commit e9af28e6a3
3 changed files with 10 additions and 10 deletions
--- a/diffsynth/pipelines/stable_diffusion_video.py
+++ b/diffsynth/pipelines/stable_diffusion_video.py
@@ -89,7 +89,7 @@ class SDVideoPipeline(torch.nn.Module):
        controlnet_units = []
        for config in controlnet_config_units:
            controlnet_unit = ControlNetUnit(
-                Annotator(config.processor_id),
+                Annotator(config.processor_id, device=self.device),
                model_manager.get_model_with_model_path(config.model_path),
                config.scale
            )