diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-14 22:03:01 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-14 22:03:01 +0100 |
| commit | fc11c86142915d6c3935d28a3321b3ae91b613ef (patch) | |
| tree | 5d2c84b1ff32e779db868da1248ed24a97cde3c2 /trainer/ti.py | |
| parent | WIP: Modularization ("free(): invalid pointer" my ass) (diff) | |
| download | textual-inversion-diff-fc11c86142915d6c3935d28a3321b3ae91b613ef.tar.gz textual-inversion-diff-fc11c86142915d6c3935d28a3321b3ae91b613ef.tar.bz2 textual-inversion-diff-fc11c86142915d6c3935d28a3321b3ae91b613ef.zip | |
Update
Diffstat (limited to 'trainer/ti.py')
| -rw-r--r-- | trainer/ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/trainer/ti.py b/trainer/ti.py index 15cf747..388acd3 100644 --- a/trainer/ti.py +++ b/trainer/ti.py | |||
| @@ -42,7 +42,7 @@ class TextualInversionCheckpointer(Checkpointer): | |||
| 42 | checkpoints_path.joinpath(f"{slugify(token)}_{step}_{postfix}.bin") | 42 | checkpoints_path.joinpath(f"{slugify(token)}_{step}_{postfix}.bin") |
| 43 | ) | 43 | ) |
| 44 | 44 | ||
| 45 | @torch.inference_mode() | 45 | @torch.no_grad() |
| 46 | def save_samples(self, step): | 46 | def save_samples(self, step): |
| 47 | ema_context = self.ema_embeddings.apply_temporary( | 47 | ema_context = self.ema_embeddings.apply_temporary( |
| 48 | self.text_encoder.text_model.embeddings.temp_token_embedding.parameters() | 48 | self.text_encoder.text_model.embeddings.temp_token_embedding.parameters() |