diff options
| author | Volpeon <git@volpeon.ink> | 2023-03-04 09:46:41 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-03-04 09:46:41 +0100 |
| commit | 73910b7f55244ce787fc6a3e6af09240ef0cdfd3 (patch) | |
| tree | 3ef927578fc54b59ab6ff1bd00c3f804c0b9a7bf /train_lora.py | |
| parent | Pipeline: Improved initial image generation (diff) | |
| download | textual-inversion-diff-73910b7f55244ce787fc6a3e6af09240ef0cdfd3.tar.gz textual-inversion-diff-73910b7f55244ce787fc6a3e6af09240ef0cdfd3.tar.bz2 textual-inversion-diff-73910b7f55244ce787fc6a3e6af09240ef0cdfd3.zip | |
Pipeline: Perlin noise for init image
Diffstat (limited to 'train_lora.py')
| -rw-r--r-- | train_lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_lora.py b/train_lora.py index e213e3d..6e72376 100644 --- a/train_lora.py +++ b/train_lora.py | |||
| @@ -15,7 +15,7 @@ from slugify import slugify | |||
| 15 | from diffusers.loaders import AttnProcsLayers | 15 | from diffusers.loaders import AttnProcsLayers |
| 16 | from diffusers.models.cross_attention import LoRAXFormersCrossAttnProcessor, LoRACrossAttnProcessor | 16 | from diffusers.models.cross_attention import LoRAXFormersCrossAttnProcessor, LoRACrossAttnProcessor |
| 17 | 17 | ||
| 18 | from util import load_config, load_embeddings_from_dir | 18 | from util.files import load_config, load_embeddings_from_dir |
| 19 | from data.csv import VlpnDataModule, keyword_filter | 19 | from data.csv import VlpnDataModule, keyword_filter |
| 20 | from training.functional import train, get_models | 20 | from training.functional import train, get_models |
| 21 | from training.lr import plot_metrics | 21 | from training.lr import plot_metrics |