prepare for latent diffusion in the first DDPM of the cascade in the Decoder

2026-02-23 14:24:28 +01:00 · 2022-04-21 17:54:31 -07:00
parent 0b4ec34efb
commit ad17c69ab6
5 changed files with 538 additions and 16 deletions
--- a/setup.py
+++ b/setup.py
@@ -30,6 +30,7 @@ setup(
    'torch>=1.10',
    'torchvision',
    'tqdm',
+    'vector-quantize-pytorch',
    'x-clip>=0.4.4',
    'youtokentome'
  ],