Update autoencoder.py

This commit is contained in:
smthemex
2024-08-11 16:23:44 +08:00
committed by GitHub
parent a562b5a6b6
commit 13fd77f9b8
+1 -1
View File
@@ -8,7 +8,7 @@ from taming.modules.vqvae.quantize import VectorQuantizer
from .model import Encoder, Decoder
from .distributions import DiagonalGaussianDistribution
from customnet_util import instantiate_from_config
from .customnet_util import instantiate_from_config
class VQModel(pl.LightningModule):