diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-11 21:54:10 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-11 21:54:10 +0100 |
| commit | 92113e3a7c9cfda2bc2f6cc0fa5b1234505f145d (patch) | |
| tree | 7c032c88126b15e2a9eb13ccc4a8293e8d660f29 /training | |
| parent | Better defaults (diff) | |
| download | textual-inversion-diff-92113e3a7c9cfda2bc2f6cc0fa5b1234505f145d.tar.gz textual-inversion-diff-92113e3a7c9cfda2bc2f6cc0fa5b1234505f145d.tar.bz2 textual-inversion-diff-92113e3a7c9cfda2bc2f6cc0fa5b1234505f145d.zip | |
TI: Use grad clipping from LoRA #104
Diffstat (limited to 'training')
| -rw-r--r-- | training/common.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/common.py b/training/common.py index 67c2ab6..0b2ae44 100644 --- a/training/common.py +++ b/training/common.py | |||
| @@ -58,7 +58,7 @@ def generate_class_images( | |||
| 58 | torch.cuda.empty_cache() | 58 | torch.cuda.empty_cache() |
| 59 | 59 | ||
| 60 | 60 | ||
| 61 | def run_model( | 61 | def loss_step( |
| 62 | vae: AutoencoderKL, | 62 | vae: AutoencoderKL, |
| 63 | noise_scheduler: DDPMScheduler, | 63 | noise_scheduler: DDPMScheduler, |
| 64 | unet: UNet2DConditionModel, | 64 | unet: UNet2DConditionModel, |