feat(worker): add num_inference_steps to I2I, I2V and Upscale pipelines

This commit adds the `num_inference_steps` to the I2I, I2V and Upscale
pipelines in the worker codebase.

runner/gen_openapi.py

@@ -1,12 +1,18 @@
 import argparse
+import copy
 import json
 import os
-import copy
 
 import yaml
 from app.main import app, use_route_names_as_operation_ids
-from app.routes import health, image_to_image, image_to_video, text_to_image, upscale, audio_to_text
-
+from app.routes import (
+    audio_to_text,
+    health,
+    image_to_image,
+    image_to_video,
+    text_to_image,
+    upscale,
+)
 from fastapi.openapi.utils import get_openapi
 
 # Specify Endpoints for OpenAPI schema generation.

runner/openapi.json

@@ -488,7 +488,7 @@
                     "num_inference_steps": {
                         "type": "integer",
                         "title": "Num Inference Steps",
-                        "default": 25
+                        "default": 100
                     },
                     "num_images_per_prompt": {
                         "type": "integer",
@@ -589,7 +589,7 @@
                     "num_inference_steps": {
                         "type": "integer",
                         "title": "Num Inference Steps",
-                        "default": 50
+                        "default": 75
                     }
                 },
                 "type": "object",

worker/docker.go

@@ -177,6 +177,7 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
 		},
 		ExposedPorts: nat.PortSet{
 			containerPort: struct{}{},
+			"5678/tcp":    struct{}{},
 		},
 		Labels: map[string]string{
 			containerCreatorLabel: containerCreator,
@@ -205,6 +206,12 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
 					HostPort: containerHostPort,
 				},
 			},
+			"5678/tcp": []nat.PortBinding{
+				{
+					HostIP:   "0.0.0.0",
+					HostPort: "5678",
+				},
+			},
 		},
 	}
 

worker/multipart.go

@@ -75,6 +75,11 @@ func NewImageToImageMultipartWriter(w io.Writer, req ImageToImageMultipartReques
 			return nil, err
 		}
 	}
+	if req.NumInferenceSteps != nil {
+		if err := mw.WriteField("num_inference_steps", strconv.Itoa(*req.NumInferenceSteps)); err != nil {
+			return nil, err
+		}
+	}
 
 	if err := mw.Close(); err != nil {
 		return nil, err
@@ -142,6 +147,11 @@ func NewImageToVideoMultipartWriter(w io.Writer, req ImageToVideoMultipartReques
 			return nil, err
 		}
 	}
+	if req.NumInferenceSteps != nil {
+		if err := mw.WriteField("num_inference_steps", strconv.Itoa(*req.NumInferenceSteps)); err != nil {
+			return nil, err
+		}
+	}
 
 	if err := mw.Close(); err != nil {
 		return nil, err
@@ -187,6 +197,11 @@ func NewUpscaleMultipartWriter(w io.Writer, req UpscaleMultipartRequestBody) (*m
 			return nil, err
 		}
 	}
+	if req.NumInferenceSteps != nil {
+		if err := mw.WriteField("num_inference_steps", strconv.Itoa(*req.NumInferenceSteps)); err != nil {
+			return nil, err
+		}
+	}
 
 	if err := mw.Close(); err != nil {
 		return nil, err