• VQ-VAE
• ..
• main.py
• model_attn.py
• model_soft.py
• utils.py