diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-30 09:36:55 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-05-30 09:47:57 +0000 |
| commit | 5fcdaa6a7f19d083a6393cc0d2b933ff5080f5b3 (patch) | |
| tree | 8a4641e83b0a78b0a0bb210d70cee247189a85b6 /extensions-builtin/LDSR/sd_hijack_autoencoder.py | |
| parent | b957dcfece29c84ac0cfcd5a69475ff8684c531f (diff) | |
| download | stable-diffusion-webui-gfx803-5fcdaa6a7f19d083a6393cc0d2b933ff5080f5b3.tar.gz stable-diffusion-webui-gfx803-5fcdaa6a7f19d083a6393cc0d2b933ff5080f5b3.tar.bz2 stable-diffusion-webui-gfx803-5fcdaa6a7f19d083a6393cc0d2b933ff5080f5b3.zip | |
Vendor in the single module used from taming_transformers; remove taming_transformers dependency
(and fix the two ruff complaints)
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_autoencoder.py')
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_autoencoder.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_autoencoder.py b/extensions-builtin/LDSR/sd_hijack_autoencoder.py index 81c5101b..27a86e13 100644 --- a/extensions-builtin/LDSR/sd_hijack_autoencoder.py +++ b/extensions-builtin/LDSR/sd_hijack_autoencoder.py @@ -10,7 +10,7 @@ from contextlib import contextmanager from torch.optim.lr_scheduler import LambdaLR from ldm.modules.ema import LitEma -from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer +from vqvae_quantize import VectorQuantizer2 as VectorQuantizer from ldm.modules.diffusionmodules.model import Encoder, Decoder from ldm.util import instantiate_from_config |