diff options
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() |