From 13fd77f9b8e8f1c3d4a84bb834086986d1c96647 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sun, 11 Aug 2024 16:23:44 +0800
Subject: [PATCH] Update autoencoder.py

---
 custom_net/autoencoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_net/autoencoder.py b/custom_net/autoencoder.py
index bed2692..8634d6a 100644
--- a/custom_net/autoencoder.py
+++ b/custom_net/autoencoder.py
@@ -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):