diff --git a/model/model/v1alpha/task_image_to_image_pb2.py b/model/model/v1alpha/task_image_to_image_pb2.py
index 254d9ad..57fdac4 100644
--- a/model/model/v1alpha/task_image_to_image_pb2.py
+++ b/model/model/v1alpha/task_image_to_image_pb2.py
@@ -15,7 +15,7 @@
 from model.model.v1alpha import common_pb2 as model_dot_model_dot_v1alpha_dot_common__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-model/model/v1alpha/task_image_to_image.proto\x12\x13model.model.v1alpha\x1a\x1fgoogle/api/field_behavior.proto\x1a model/model/v1alpha/common.proto\"\xc7\x03\n\x11ImageToImageInput\x12*\n\x10prompt_image_url\x18\x01 \x01(\tH\x00R\x0epromptImageUrl\x12\x30\n\x13prompt_image_base64\x18\x02 \x01(\tH\x00R\x11promptImageBase64\x12!\n\x06prompt\x18\x03 \x01(\tB\x04\xe2\x41\x01\x02H\x01R\x06prompt\x88\x01\x01\x12\x1f\n\x05steps\x18\x04 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x02R\x05steps\x88\x01\x01\x12&\n\tcfg_scale\x18\x05 \x01(\x02\x42\x04\xe2\x41\x01\x01H\x03R\x08\x63\x66gScale\x88\x01\x01\x12\x1d\n\x04seed\x18\x06 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x04R\x04seed\x88\x01\x01\x12#\n\x07samples\x18\x07 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x05R\x07samples\x88\x01\x01\x12S\n\x0c\x65xtra_params\x18\x08 \x01(\x0b\x32%.model.model.v1alpha.ExtraParamObjectB\x04\xe2\x41\x01\x01H\x06R\x0b\x65xtraParams\x88\x01\x01\x42\x06\n\x04typeB\t\n\x07_promptB\x08\n\x06_stepsB\x0c\n\n_cfg_scaleB\x07\n\x05_seedB\n\n\x08_samplesB\x0f\n\r_extra_params\"2\n\x12ImageToImageOutput\x12\x1c\n\x06images\x18\x01 \x03(\tB\x04\xe2\x41\x01\x03R\x06imagesB\xe2\x01\n\x17\x63om.model.model.v1alphaB\x15TaskImageToImageProtoP\x01ZBgithub.com/instill-ai/protogen-go/model/model/v1alpha;modelv1alpha\xa2\x02\x03MMX\xaa\x02\x13Model.Model.V1alpha\xca\x02\x13Model\\Model\\V1alpha\xe2\x02\x1fModel\\Model\\V1alpha\\GPBMetadata\xea\x02\x15Model::Model::V1alphab\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-model/model/v1alpha/task_image_to_image.proto\x12\x13model.model.v1alpha\x1a\x1fgoogle/api/field_behavior.proto\x1a model/model/v1alpha/common.proto\"\xb1\x03\n\x11ImageToImageInput\x12*\n\x10prompt_image_url\x18\x01 \x01(\tH\x00R\x0epromptImageUrl\x12\x30\n\x13prompt_image_base64\x18\x02 \x01(\tH\x00R\x11promptImageBase64\x12!\n\x06prompt\x18\x03 \x01(\tB\x04\xe2\x41\x01\x02H\x01R\x06prompt\x88\x01\x01\x12\x1f\n\x05steps\x18\x04 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x02R\x05steps\x88\x01\x01\x12&\n\tcfg_scale\x18\x05 \x01(\x02\x42\x04\xe2\x41\x01\x01H\x03R\x08\x63\x66gScale\x88\x01\x01\x12\x1d\n\x04seed\x18\x06 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x04R\x04seed\x88\x01\x01\x12#\n\x07samples\x18\x07 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x05R\x07samples\x88\x01\x01\x12N\n\x0c\x65xtra_params\x18\x08 \x03(\x0b\x32%.model.model.v1alpha.ExtraParamObjectB\x04\xe2\x41\x01\x01R\x0b\x65xtraParamsB\x06\n\x04typeB\t\n\x07_promptB\x08\n\x06_stepsB\x0c\n\n_cfg_scaleB\x07\n\x05_seedB\n\n\x08_samples\"2\n\x12ImageToImageOutput\x12\x1c\n\x06images\x18\x01 \x03(\tB\x04\xe2\x41\x01\x03R\x06imagesB\xe2\x01\n\x17\x63om.model.model.v1alphaB\x15TaskImageToImageProtoP\x01ZBgithub.com/instill-ai/protogen-go/model/model/v1alpha;modelv1alpha\xa2\x02\x03MMX\xaa\x02\x13Model.Model.V1alpha\xca\x02\x13Model\\Model\\V1alpha\xe2\x02\x1fModel\\Model\\V1alpha\\GPBMetadata\xea\x02\x15Model::Model::V1alphab\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -39,7 +39,7 @@
   _IMAGETOIMAGEOUTPUT.fields_by_name['images']._options = None
   _IMAGETOIMAGEOUTPUT.fields_by_name['images']._serialized_options = b'\342A\001\003'
   _globals['_IMAGETOIMAGEINPUT']._serialized_start=138
-  _globals['_IMAGETOIMAGEINPUT']._serialized_end=593
-  _globals['_IMAGETOIMAGEOUTPUT']._serialized_start=595
-  _globals['_IMAGETOIMAGEOUTPUT']._serialized_end=645
+  _globals['_IMAGETOIMAGEINPUT']._serialized_end=571
+  _globals['_IMAGETOIMAGEOUTPUT']._serialized_start=573
+  _globals['_IMAGETOIMAGEOUTPUT']._serialized_end=623
 # @@protoc_insertion_point(module_scope)
diff --git a/model/model/v1alpha/task_image_to_image_pb2.pyi b/model/model/v1alpha/task_image_to_image_pb2.pyi
index c8d97b3..06ebba7 100644
--- a/model/model/v1alpha/task_image_to_image_pb2.pyi
+++ b/model/model/v1alpha/task_image_to_image_pb2.pyi
@@ -47,7 +47,7 @@ class ImageToImageInput(google.protobuf.message.Message):
     samples: builtins.int
     """The number of generated samples, default is 1"""
     @property
-    def extra_params(self) -> model.model.v1alpha.common_pb2.ExtraParamObject:
+    def extra_params(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[model.model.v1alpha.common_pb2.ExtraParamObject]:
         """The extra parameters"""
     def __init__(
         self,
@@ -59,15 +59,13 @@ class ImageToImageInput(google.protobuf.message.Message):
         cfg_scale: builtins.float | None = ...,
         seed: builtins.int | None = ...,
         samples: builtins.int | None = ...,
-        extra_params: model.model.v1alpha.common_pb2.ExtraParamObject | None = ...,
+        extra_params: collections.abc.Iterable[model.model.v1alpha.common_pb2.ExtraParamObject] | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_extra_params", b"_extra_params", "_prompt", b"_prompt", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "extra_params", b"extra_params", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_extra_params", b"_extra_params", "_prompt", b"_prompt", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "extra_params", b"extra_params", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_prompt", b"_prompt", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_prompt", b"_prompt", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "extra_params", b"extra_params", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_cfg_scale", b"_cfg_scale"]) -> typing_extensions.Literal["cfg_scale"] | None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_extra_params", b"_extra_params"]) -> typing_extensions.Literal["extra_params"] | None: ...
-    @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_prompt", b"_prompt"]) -> typing_extensions.Literal["prompt"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_samples", b"_samples"]) -> typing_extensions.Literal["samples"] | None: ...
diff --git a/model/model/v1alpha/task_text_to_image_pb2.py b/model/model/v1alpha/task_text_to_image_pb2.py
index f9980c5..5e3c5d1 100644
--- a/model/model/v1alpha/task_text_to_image_pb2.py
+++ b/model/model/v1alpha/task_text_to_image_pb2.py
@@ -15,7 +15,7 @@
 from model.model.v1alpha import common_pb2 as model_dot_model_dot_v1alpha_dot_common__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,model/model/v1alpha/task_text_to_image.proto\x12\x13model.model.v1alpha\x1a\x1fgoogle/api/field_behavior.proto\x1a model/model/v1alpha/common.proto\"\xb6\x03\n\x10TextToImageInput\x12\x1c\n\x06prompt\x18\x01 \x01(\tB\x04\xe2\x41\x01\x02R\x06prompt\x12*\n\x10prompt_image_url\x18\x02 \x01(\tH\x00R\x0epromptImageUrl\x12\x30\n\x13prompt_image_base64\x18\x03 \x01(\tH\x00R\x11promptImageBase64\x12\x1f\n\x05steps\x18\x04 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x01R\x05steps\x88\x01\x01\x12&\n\tcfg_scale\x18\x05 \x01(\x02\x42\x04\xe2\x41\x01\x01H\x02R\x08\x63\x66gScale\x88\x01\x01\x12\x1d\n\x04seed\x18\x06 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x03R\x04seed\x88\x01\x01\x12#\n\x07samples\x18\x07 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x04R\x07samples\x88\x01\x01\x12S\n\x0c\x65xtra_params\x18\x08 \x01(\x0b\x32%.model.model.v1alpha.ExtraParamObjectB\x04\xe2\x41\x01\x01H\x05R\x0b\x65xtraParams\x88\x01\x01\x42\x06\n\x04typeB\x08\n\x06_stepsB\x0c\n\n_cfg_scaleB\x07\n\x05_seedB\n\n\x08_samplesB\x0f\n\r_extra_params\"1\n\x11TextToImageOutput\x12\x1c\n\x06images\x18\x01 \x03(\tB\x04\xe2\x41\x01\x03R\x06imagesB\xe1\x01\n\x17\x63om.model.model.v1alphaB\x14TaskTextToImageProtoP\x01ZBgithub.com/instill-ai/protogen-go/model/model/v1alpha;modelv1alpha\xa2\x02\x03MMX\xaa\x02\x13Model.Model.V1alpha\xca\x02\x13Model\\Model\\V1alpha\xe2\x02\x1fModel\\Model\\V1alpha\\GPBMetadata\xea\x02\x15Model::Model::V1alphab\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,model/model/v1alpha/task_text_to_image.proto\x12\x13model.model.v1alpha\x1a\x1fgoogle/api/field_behavior.proto\x1a model/model/v1alpha/common.proto\"\xa0\x03\n\x10TextToImageInput\x12\x1c\n\x06prompt\x18\x01 \x01(\tB\x04\xe2\x41\x01\x02R\x06prompt\x12*\n\x10prompt_image_url\x18\x02 \x01(\tH\x00R\x0epromptImageUrl\x12\x30\n\x13prompt_image_base64\x18\x03 \x01(\tH\x00R\x11promptImageBase64\x12\x1f\n\x05steps\x18\x04 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x01R\x05steps\x88\x01\x01\x12&\n\tcfg_scale\x18\x05 \x01(\x02\x42\x04\xe2\x41\x01\x01H\x02R\x08\x63\x66gScale\x88\x01\x01\x12\x1d\n\x04seed\x18\x06 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x03R\x04seed\x88\x01\x01\x12#\n\x07samples\x18\x07 \x01(\x05\x42\x04\xe2\x41\x01\x01H\x04R\x07samples\x88\x01\x01\x12N\n\x0c\x65xtra_params\x18\x08 \x03(\x0b\x32%.model.model.v1alpha.ExtraParamObjectB\x04\xe2\x41\x01\x01R\x0b\x65xtraParamsB\x06\n\x04typeB\x08\n\x06_stepsB\x0c\n\n_cfg_scaleB\x07\n\x05_seedB\n\n\x08_samples\"1\n\x11TextToImageOutput\x12\x1c\n\x06images\x18\x01 \x03(\tB\x04\xe2\x41\x01\x03R\x06imagesB\xe1\x01\n\x17\x63om.model.model.v1alphaB\x14TaskTextToImageProtoP\x01ZBgithub.com/instill-ai/protogen-go/model/model/v1alpha;modelv1alpha\xa2\x02\x03MMX\xaa\x02\x13Model.Model.V1alpha\xca\x02\x13Model\\Model\\V1alpha\xe2\x02\x1fModel\\Model\\V1alpha\\GPBMetadata\xea\x02\x15Model::Model::V1alphab\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -39,7 +39,7 @@
   _TEXTTOIMAGEOUTPUT.fields_by_name['images']._options = None
   _TEXTTOIMAGEOUTPUT.fields_by_name['images']._serialized_options = b'\342A\001\003'
   _globals['_TEXTTOIMAGEINPUT']._serialized_start=137
-  _globals['_TEXTTOIMAGEINPUT']._serialized_end=575
-  _globals['_TEXTTOIMAGEOUTPUT']._serialized_start=577
-  _globals['_TEXTTOIMAGEOUTPUT']._serialized_end=626
+  _globals['_TEXTTOIMAGEINPUT']._serialized_end=553
+  _globals['_TEXTTOIMAGEOUTPUT']._serialized_start=555
+  _globals['_TEXTTOIMAGEOUTPUT']._serialized_end=604
 # @@protoc_insertion_point(module_scope)
diff --git a/model/model/v1alpha/task_text_to_image_pb2.pyi b/model/model/v1alpha/task_text_to_image_pb2.pyi
index d0bd3b9..53ccaac 100644
--- a/model/model/v1alpha/task_text_to_image_pb2.pyi
+++ b/model/model/v1alpha/task_text_to_image_pb2.pyi
@@ -47,7 +47,7 @@ class TextToImageInput(google.protobuf.message.Message):
     samples: builtins.int
     """The number of generated samples, default is 1"""
     @property
-    def extra_params(self) -> model.model.v1alpha.common_pb2.ExtraParamObject:
+    def extra_params(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[model.model.v1alpha.common_pb2.ExtraParamObject]:
         """The extra parameters"""
     def __init__(
         self,
@@ -59,15 +59,13 @@ class TextToImageInput(google.protobuf.message.Message):
         cfg_scale: builtins.float | None = ...,
         seed: builtins.int | None = ...,
         samples: builtins.int | None = ...,
-        extra_params: model.model.v1alpha.common_pb2.ExtraParamObject | None = ...,
+        extra_params: collections.abc.Iterable[model.model.v1alpha.common_pb2.ExtraParamObject] | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_extra_params", b"_extra_params", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "extra_params", b"extra_params", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_extra_params", b"_extra_params", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "extra_params", b"extra_params", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_cfg_scale", b"_cfg_scale", "_samples", b"_samples", "_seed", b"_seed", "_steps", b"_steps", "cfg_scale", b"cfg_scale", "extra_params", b"extra_params", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "samples", b"samples", "seed", b"seed", "steps", b"steps", "type", b"type"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_cfg_scale", b"_cfg_scale"]) -> typing_extensions.Literal["cfg_scale"] | None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_extra_params", b"_extra_params"]) -> typing_extensions.Literal["extra_params"] | None: ...
-    @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_samples", b"_samples"]) -> typing_extensions.Literal["samples"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_seed", b"_seed"]) -> typing_extensions.Literal["seed"] | None: ...