Add batch size in the Constructor.

[prajinkhadka]

Using 10 as batch size was throwing an issue :

"error": {
"message": "{'task_type': 'text2im', 'prompt': {'caption': 'portal to another dimension, digital art', 'batch_size': 10}} is not valid under any of the given schemas",
"type": "invalid_request_error",
"param": null,
"code": null
}
}

[ezzcodeezzlife]

thanks 🔥