All changes implemented, current run happening. Link to wandb run in comments. (#43)

* Train DiffusionPrior with pre-computed embeddings This is in response to https://github.com/lucidrains/DALLE2-pytorch/issues/29 - more metrics will get added.
2026-02-19 02:54:25 +01:00 · 2022-05-02 00:16:59 +05:30
parent ad8d7a368b
commit 53ce6dfdf6
2 changed files with 201 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,8 @@ setup(
    'tqdm',
    'vector-quantize-pytorch',
    'x-clip>=0.5.1',
-    'youtokentome'
+    'youtokentome',
+    'embedding-reader'
  ],
  classifiers=[
    'Development Status :: 4 - Beta',