diff --git a/src/api/routes/images.ts b/src/api/routes/images.ts
index c941913..4988a14 100644
--- a/src/api/routes/images.ts
+++ b/src/api/routes/images.ts
@@ -61,12 +61,12 @@ export default {
   post: {
     "/generations": async (request: Request) => {
       request
-        // .validate("body.model", v => _.isUndefined(v) || _.isString(v))
+        .validate("body.model", v => _.isUndefined(v) || _.isString(v))
         .validate("body.task_id", _.isString)
         .validate("body.prompt", _.isString)
         // .validate("body.negative_prompt", v => _.isUndefined(v) || _.isString(v))
-        .validate("body.width", v => _.isUndefined(v) || _.isFinite(v))
-        .validate("body.height", v => _.isUndefined(v) || _.isFinite(v))
+        .validate("body.width", v => _.isUndefined(v) || _.isFinite(v) || _.isString(v))
+        .validate("body.height", v => _.isUndefined(v) || _.isFinite(v) || _.isString(v))
         // .validate("body.sample_strength", v => _.isUndefined(v) || _.isFinite(v))
         .validate("body.response_format", v => _.isUndefined(v) || _.isString(v))
         .validate("headers.authorization", _.isString);