diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 04:52:45 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 04:52:45 +0000 |
| commit | 762265eab58cdb8f2d6398769bab43d8b8db0075 (patch) | |
| tree | 7766901f867d84cf0662b2ae0d5bc55c673a08e8 /extensions-builtin/LDSR | |
| parent | a617d6488275a58da0627b3fed5f53593b2eb8b2 (diff) | |
| download | stable-diffusion-webui-gfx803-762265eab58cdb8f2d6398769bab43d8b8db0075.tar.gz stable-diffusion-webui-gfx803-762265eab58cdb8f2d6398769bab43d8b8db0075.tar.bz2 stable-diffusion-webui-gfx803-762265eab58cdb8f2d6398769bab43d8b8db0075.zip | |
autofixes from ruff
Diffstat (limited to 'extensions-builtin/LDSR')
| -rw-r--r-- | extensions-builtin/LDSR/ldsr_model_arch.py | 1 | ||||
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_autoencoder.py | 2 |
2 files changed, 1 insertions, 2 deletions
diff --git a/extensions-builtin/LDSR/ldsr_model_arch.py b/extensions-builtin/LDSR/ldsr_model_arch.py index bc11cc6e..2339de7f 100644 --- a/extensions-builtin/LDSR/ldsr_model_arch.py +++ b/extensions-builtin/LDSR/ldsr_model_arch.py @@ -110,7 +110,6 @@ class LDSR: diffusion_steps = int(steps) eta = 1.0 - down_sample_method = 'Lanczos' gc.collect() if torch.cuda.is_available: diff --git a/extensions-builtin/LDSR/sd_hijack_autoencoder.py b/extensions-builtin/LDSR/sd_hijack_autoencoder.py index 8e03c7f8..db2231dd 100644 --- a/extensions-builtin/LDSR/sd_hijack_autoencoder.py +++ b/extensions-builtin/LDSR/sd_hijack_autoencoder.py @@ -165,7 +165,7 @@ class VQModel(pl.LightningModule): def validation_step(self, batch, batch_idx): log_dict = self._validation_step(batch, batch_idx) with self.ema_scope(): - log_dict_ema = self._validation_step(batch, batch_idx, suffix="_ema") + self._validation_step(batch, batch_idx, suffix="_ema") return log_dict def _validation_step(self, batch, batch_idx, suffix=""): |