Unverified Commit e6a84922 authored by M. Tolga Cangöz's avatar M. Tolga Cangöz Committed by GitHub
Browse files

Use PyTorch's conventional inplace functions (#7332)


Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent ad0308b3
......@@ -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(
......
......@@ -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(
......
......@@ -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(
......
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment