grid_size

hydra_node/models.py

@@ -140,7 +140,7 @@ class DalleMiniModel(nn.Module):
         images = self.model.generate_images(
             text=request.prompt,
             seed=seed,
-            grid_size=4,
+            grid_size=request.grid_size,
             is_seamless=False,
             temperature=request.temp,
             top_k=request.top_k,