Vendor in the single module used from taming_transformers; remove taming_transformers dependency

(and fix the two ruff complaints)
This commit is contained in:
Aarni Koskela
2023-05-30 12:36:55 +03:00
parent b957dcfece
commit 5fcdaa6a7f
5 changed files with 148 additions and 6 deletions

View File

@@ -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