vq_gan_3d/model/__init__.py

from vq_gan_3d.model.vqgan import VQGAN, load_VQGAN
from vq_gan_3d.model.codebook import Codebook
from vq_gan_3d.model.lpips import LPIPS