diff --git a/tests/pipelines/controlnet/test_controlnet.py b/tests/pipelines/controlnet/test_controlnet.py index 114a36b37f..cc0696bae9 100644 --- a/tests/pipelines/controlnet/test_controlnet.py +++ b/tests/pipelines/controlnet/test_controlnet.py @@ -302,7 +302,7 @@ class StableDiffusionMultiControlNetPipelineFastTests( def init_weights(m): if isinstance(m, torch.nn.Conv2d): - torch.nn.init.normal(m.weight) + torch.nn.init.normal_(m.weight) m.bias.data.fill_(1.0) controlnet1 = ControlNetModel( @@ -519,7 +519,7 @@ class StableDiffusionMultiControlNetOneModelPipelineFastTests( def init_weights(m): if isinstance(m, torch.nn.Conv2d): - torch.nn.init.normal(m.weight) + torch.nn.init.normal_(m.weight) m.bias.data.fill_(1.0) controlnet = ControlNetModel( diff --git a/tests/pipelines/controlnet/test_controlnet_img2img.py b/tests/pipelines/controlnet/test_controlnet_img2img.py index 89e2b3803d..46821a51b7 100644 --- a/tests/pipelines/controlnet/test_controlnet_img2img.py +++ b/tests/pipelines/controlnet/test_controlnet_img2img.py @@ -210,7 +210,7 @@ class StableDiffusionMultiControlNetPipelineFastTests( def init_weights(m): if isinstance(m, torch.nn.Conv2d): - torch.nn.init.normal(m.weight) + torch.nn.init.normal_(m.weight) m.bias.data.fill_(1.0) controlnet1 = ControlNetModel( diff --git a/tests/pipelines/controlnet/test_controlnet_inpaint.py b/tests/pipelines/controlnet/test_controlnet_inpaint.py index 67e0da4de9..32ae8d125a 100644 --- a/tests/pipelines/controlnet/test_controlnet_inpaint.py +++ b/tests/pipelines/controlnet/test_controlnet_inpaint.py @@ -273,7 +273,7 @@ class MultiControlNetInpaintPipelineFastTests( def init_weights(m): if isinstance(m, torch.nn.Conv2d): - torch.nn.init.normal(m.weight) + torch.nn.init.normal_(m.weight) m.bias.data.fill_(1.0) controlnet1 = ControlNetModel( diff --git a/tests/pipelines/controlnet/test_controlnet_sdxl.py b/tests/pipelines/controlnet/test_controlnet_sdxl.py index c82ce6c39c..e06f228a7d 100644 --- a/tests/pipelines/controlnet/test_controlnet_sdxl.py +++ b/tests/pipelines/controlnet/test_controlnet_sdxl.py @@ -503,7 +503,7 @@ class StableDiffusionXLMultiControlNetPipelineFastTests( def init_weights(m): if isinstance(m, torch.nn.Conv2d): - torch.nn.init.normal(m.weight) + torch.nn.init.normal_(m.weight) m.bias.data.fill_(1.0) controlnet1 = ControlNetModel( @@ -708,7 +708,7 @@ class StableDiffusionXLMultiControlNetOneModelPipelineFastTests( def init_weights(m): if isinstance(m, torch.nn.Conv2d): - torch.nn.init.normal(m.weight) + torch.nn.init.normal_(m.weight) m.bias.data.fill_(1.0) controlnet = ControlNetModel(