fix readme and a small bug in DALLE2 class

2025-12-19 01:34:19 +01:00 · 2022-07-28 08:33:51 -07:00
parent 07abfcf45b
commit 36fb46a95e
3 changed files with 3 additions and 2 deletions
--- a/README.md
+++ b/README.md
@@ -371,6 +371,7 @@ loss.backward()
 unet1 = Unet(
    dim = 128,
    image_embed_dim = 512,
+    text_embed_dim = 512,
    cond_dim = 128,
    channels = 3,
    dim_mults=(1, 2, 4, 8),