Erasing-Concepts-In-Diffusion

Runtime error

JadenFK commited on Mar 17, 2023

Commit

ffae5f8

•

1 Parent(s): f9f0452

bug

Files changed (1) hide show

stable_diffusion/ldm/models/autoencoder.py CHANGED Viewed

@@ -3,7 +3,7 @@ import pytorch_lightning as pl
 import torch.nn.functional as F
 from contextlib import contextmanager
-from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
 from ldm.modules.diffusionmodules.model import Encoder, Decoder
 from ldm.modules.distributions.distributions import DiagonalGaussianDistribution

 import torch.nn.functional as F
 from contextlib import contextmanager
+from taming.modules.vqvae.quantize import VectorQuantizer
 from ldm.modules.diffusionmodules.model import Encoder, Decoder
 from ldm.modules.distributions.distributions import DiagonalGaussianDistribution