Spaces:
Running
on
A10G
Running
on
A10G
Update ldm/models/autoencoder.py
Browse files
ldm/models/autoencoder.py
CHANGED
|
@@ -3,6 +3,8 @@ import pytorch_lightning as pl
|
|
| 3 |
import torch.nn.functional as F
|
| 4 |
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
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from contextlib import contextmanager
|
| 5 |
|
| 6 |
+
import sys
|
| 7 |
+
sys.path.append('../../../taming-transformers')
|
| 8 |
from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
|
| 9 |
|
| 10 |
from ldm.modules.diffusionmodules.model import Encoder, Decoder
|