diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2024-01-05 16:38:05 +0800 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2024-01-05 16:38:05 +0800 |
| commit | 44744d6005da5e424267698ee3279caa597dfebc (patch) | |
| tree | a86ffd65a13d410af5912253008cbfea6a2fdbe0 /extensions-builtin | |
| parent | 18ca987c92f52690daec43a6c67363c341bb6008 (diff) | |
linting
Diffstat (limited to 'extensions-builtin')
| -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):
|