Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -39,7 +39,8 @@ class MinDalle:
         self.dtype = dtype
         self.is_verbose = is_verbose
         self.text_token_count = 64
-        self.layer_count = 24 if is_mega else 12
+        #Changed
+        self.layer_count = 24 if is_mega else 6
         self.attention_head_count = 32 if is_mega else 16
         self.embed_count = 2048 if is_mega else 1024
         self.glu_embed_count = 4096 if is_mega else 2730