ysharma HF staff commited on
Commit
f7eb34d
1 Parent(s): 80c7c70
Files changed (1) hide show
  1. autoencoder.py +3 -3
autoencoder.py CHANGED
@@ -5,10 +5,10 @@ from contextlib import contextmanager
5
 
6
  from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
7
 
8
- from ldm.modules.diffusionmodules.model import Encoder, Decoder
9
- from ldm.modules.distributions.distributions import DiagonalGaussianDistribution
10
 
11
- from ldm.util import instantiate_from_config
12
 
13
 
14
  class VQModel(pl.LightningModule):
 
5
 
6
  from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
7
 
8
+ from model import Encoder, Decoder
9
+ from distributions import DiagonalGaussianDistribution
10
 
11
+ from util import instantiate_from_config
12
 
13
 
14
  class VQModel(pl.LightningModule):