diff options
| -rw-r--r-- | extensions-builtin/Lora/network.py | 3 |
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):
|