File size: 135 Bytes
913ec88
 
 
 
 
 
1
2
3
4
5
6
from typing import Tuple
import torch.nn as nn

# from .quant import VectorQuantizer2
# from .var import VAR
# from .vqvae import VQVAE