change batch size to 6

ezzcodeezzlife · ezzcodeezzlife · commit e9b8ca3d483b · 2022-06-21T14:10:26.000+02:00
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = dalle2
-version = 1.0.0
+version = 1.0.1
 author = ezzcodeezzlife
 author_email = author@example.com
 description = Use DALL·E 2 in Python
diff --git a/src/dalle2/dalle2.py b/src/dalle2/dalle2.py
@@ -16,7 +16,7 @@ def generate(self, promt):
             "task_type": "text2im",
             "prompt": {
                 "caption": promt,
-                "batch_size": 10,
+                "batch_size": 6,
             }
         }
 
