from typing import Tuple import torch.nn as nn # from .quant import VectorQuantizer2 # from .var import VAR # from .vqvae import VQVAE