AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Fix PyTorch's convention for inplace functions

Browse Source
This commit is contained in:
standardAI
2024-03-14 10:11:41 +03:00
parent 53e9aacc10
commit f65dc4afcb
4 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/pipelines/controlnet/test_controlnet.py
View File

@@ -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(

2
tests/pipelines/controlnet/test_controlnet_img2img.py
View File

@@ -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(

2
tests/pipelines/controlnet/test_controlnet_inpaint.py
View File

@@ -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(

4
tests/pipelines/controlnet/test_controlnet_sdxl.py
View File

@@ -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(