This commit is contained in:
discus0434 2022-10-19 01:00:01 +09:00
parent 6021f7a75f
commit a5611ea502
2 changed files with 9 additions and 8 deletions

1
.gitignore vendored
View file

@ -27,4 +27,3 @@ __pycache__
notification.mp3 notification.mp3
/SwinIR /SwinIR
/textual_inversion /textual_inversion
/hypernetwork

View file

@ -1,20 +1,22 @@
import csv
import datetime import datetime
import glob import glob
import html import html
import os import os
import sys import sys
import traceback import traceback
import modules.textual_inversion.dataset
import torch
import tqdm import tqdm
from einops import rearrange, repeat import csv
import torch
from ldm.util import default 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 import textual_inversion
from modules.textual_inversion.learn_schedule import LearnRateScheduler from modules.textual_inversion.learn_schedule import LearnRateScheduler
from torch import einsum
def parse_layer_structure(dim, state_dict): def parse_layer_structure(dim, state_dict):