diff --git a/tests/pipelines/controlnet/test_controlnet.py b/tests/pipelines/controlnet/test_controlnet.py index bd6330c12c40..8c391becc757 100644 --- a/tests/pipelines/controlnet/test_controlnet.py +++ b/tests/pipelines/controlnet/test_controlnet.py @@ -302,7 +302,7 @@ def get_dummy_components(self): 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 @@ def get_dummy_components(self): 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 0e04325d85b2..7496021a2c1b 100644 --- a/tests/pipelines/controlnet/test_controlnet_img2img.py +++ b/tests/pipelines/controlnet/test_controlnet_img2img.py @@ -210,7 +210,7 @@ def get_dummy_components(self): 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 4cefa4a0463c..6a4ec79d1eb2 100644 --- a/tests/pipelines/controlnet/test_controlnet_inpaint.py +++ b/tests/pipelines/controlnet/test_controlnet_inpaint.py @@ -273,7 +273,7 @@ def get_dummy_components(self): 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 c82ce6c39cca..e06f228a7d30 100644 --- a/tests/pipelines/controlnet/test_controlnet_sdxl.py +++ b/tests/pipelines/controlnet/test_controlnet_sdxl.py @@ -503,7 +503,7 @@ def get_dummy_components(self): 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 @@ def get_dummy_components(self): 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(