diff options
| -rw-r--r-- | .gitignore | 1 | ||||
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 14 |
2 files changed, 8 insertions, 7 deletions
@@ -27,4 +27,3 @@ __pycache__ notification.mp3 /SwinIR /textual_inversion -/hypernetwork diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index cadb9911..c5835bce 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -1,20 +1,22 @@ -import csv
import datetime
import glob
import html
import os
import sys
import traceback
+import tqdm
+import csv
-import modules.textual_inversion.dataset
import torch
-import tqdm
-from einops import rearrange, repeat
+
from ldm.util import default
-from modules import devices, processing, sd_models, shared
+from modules import devices, shared, processing, sd_models
+import torch
+from torch import einsum
+from einops import rearrange, repeat
+import modules.textual_inversion.dataset
from modules.textual_inversion import textual_inversion
from modules.textual_inversion.learn_schedule import LearnRateScheduler
-from torch import einsum
def parse_layer_structure(dim, state_dict):
|