diff options
author | Volpeon <git@volpeon.ink> | 2023-02-13 17:19:18 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-02-13 17:19:18 +0100 |
commit | 94b676d91382267e7429bd68362019868affd9d1 (patch) | |
tree | 513697739ab25217cbfcff630299d02b1f6e98c8 /training/functional.py | |
parent | Integrate Self-Attention-Guided (SAG) Stable Diffusion in my custom pipeline (diff) | |
download | textual-inversion-diff-94b676d91382267e7429bd68362019868affd9d1.tar.gz textual-inversion-diff-94b676d91382267e7429bd68362019868affd9d1.tar.bz2 textual-inversion-diff-94b676d91382267e7429bd68362019868affd9d1.zip |
Update
Diffstat (limited to 'training/functional.py')
-rw-r--r-- | training/functional.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/functional.py b/training/functional.py index ccbb4ad..83e70e2 100644 --- a/training/functional.py +++ b/training/functional.py | |||
@@ -129,7 +129,7 @@ def save_samples( | |||
129 | 129 | ||
130 | for pool, data, gen in datasets: | 130 | for pool, data, gen in datasets: |
131 | all_samples = [] | 131 | all_samples = [] |
132 | file_path = output_dir.joinpath(pool, f"step_{step}.jpg") | 132 | file_path = output_dir / pool / f"step_{step}.jpg" |
133 | file_path.parent.mkdir(parents=True, exist_ok=True) | 133 | file_path.parent.mkdir(parents=True, exist_ok=True) |
134 | 134 | ||
135 | batches = list(itertools.islice(itertools.cycle(data), batch_size * num_batches)) | 135 | batches = list(itertools.islice(itertools.cycle(data), batch_size * num_batches)) |