mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-08-05 03:32:37 +00:00
linting
This commit is contained in:
@@ -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):
|
||||||
|
Reference in New Issue
Block a user