mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-12-06 02:30:30 -08:00
linting
This commit is contained in:
parent
18ca987c92
commit
44744d6005
1 changed files with 1 additions and 2 deletions
|
|
@ -3,7 +3,6 @@ import os
|
||||||
from collections import namedtuple
|
from collections import namedtuple
|
||||||
import enum
|
import enum
|
||||||
|
|
||||||
import torch
|
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
import torch.nn.functional as F
|
import torch.nn.functional as F
|
||||||
|
|
||||||
|
|
@ -124,7 +123,7 @@ class NetworkModule:
|
||||||
if isinstance(self.sd_module, nn.Conv2d):
|
if isinstance(self.sd_module, nn.Conv2d):
|
||||||
self.ops = F.conv2d
|
self.ops = F.conv2d
|
||||||
self.extra_kwargs = {
|
self.extra_kwargs = {
|
||||||
'stride': self.sd_module.stride,
|
'stride': self.sd_module.stride,
|
||||||
'padding': self.sd_module.padding
|
'padding': self.sd_module.padding
|
||||||
}
|
}
|
||||||
elif isinstance(self.sd_module, nn.Linear):
|
elif isinstance(self.sd_module, nn.Linear):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue