aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/Lora/network.py
diff options
context:
space:
mode:
Diffstat (limited to 'extensions-builtin/Lora/network.py')
-rw-r--r--extensions-builtin/Lora/network.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/extensions-builtin/Lora/network.py b/extensions-builtin/Lora/network.py
index f9b571b5..b8fd9194 100644
--- a/extensions-builtin/Lora/network.py
+++ b/extensions-builtin/Lora/network.py
@@ -3,7 +3,6 @@ import os
from collections import namedtuple
import enum
-import torch
import torch.nn as nn
import torch.nn.functional as F
@@ -124,7 +123,7 @@ class NetworkModule:
if isinstance(self.sd_module, nn.Conv2d):
self.ops = F.conv2d
self.extra_kwargs = {
- 'stride': self.sd_module.stride,
+ 'stride': self.sd_module.stride,
'padding': self.sd_module.padding
}
elif isinstance(self.sd_module, nn.Linear):