inplace scatter

guiyrt · guiyrt · commit 34ab1af83eed · 2025-02-08T15:32:39.000Z
diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py
@@ -1245,10 +1245,10 @@ def __call__(
         )
 
         if isinstance(controlnet, ControlNetUnionModel):
-            control_type = torch.zeros(controlnet.config.num_control_type).scatter(0, torch.tensor(control_mode), 1)
+            control_type = torch.zeros(controlnet.config.num_control_type).scatter_(0, torch.tensor(control_mode), 1)
         elif isinstance(controlnet, MultiControlNetUnionModel):
             control_type = [
-                torch.zeros(controlnet_.config.num_control_type).scatter(0, torch.tensor(control_mode_), 1)
+                torch.zeros(controlnet_.config.num_control_type).scatter_(0, torch.tensor(control_mode_), 1)
                 for control_mode_, controlnet_ in zip(control_mode, self.controlnet.nets)
             ]
 

Original file line number	Diff line number	Diff line change
`@@ -1245,10 +1245,10 @@ def __call__(`
`1245`	`1245`	`)`
`1246`	`1246`
`1247`	`1247`	`if isinstance(controlnet, ControlNetUnionModel):`
`1248`		`- control_type = torch.zeros(controlnet.config.num_control_type).scatter(0, torch.tensor(control_mode), 1)`
	`1248`	`+ control_type = torch.zeros(controlnet.config.num_control_type).scatter_(0, torch.tensor(control_mode), 1)`
`1249`	`1249`	`elif isinstance(controlnet, MultiControlNetUnionModel):`
`1250`	`1250`	`control_type = [`
`1251`		`- torch.zeros(controlnet_.config.num_control_type).scatter(0, torch.tensor(control_mode_), 1)`
	`1251`	`+ torch.zeros(controlnet_.config.num_control_type).scatter_(0, torch.tensor(control_mode_), 1)`
`1252`	`1252`	`for control_mode_, controlnet_ in zip(control_mode, self.controlnet.nets)`
`1253`	`1253`	`]`
`1254`	`1254`