From 13e415da6fa7a89508001301318cfc53b607c61d Mon Sep 17 00:00:00 2001 From: Dayanne Fernandes Date: Mon, 30 Sep 2024 17:40:00 -0300 Subject: [PATCH] adjust postprocessing --- vision_agent/tools/tools.py | 2 +- vision_agent/tools/tools_types.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index 93ee9207..f0537254 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -422,9 +422,9 @@ def florence2_sam2_image( task=PromptTask.PHRASE_GROUNDING, tool="florencev2_fine_tuning", prompt=prompt, + postprocessing="sam2", fine_tuning=FineTuning( job_id=UUID(fine_tune_id), - postprocessing="sam2", ), ) req_data = req_data_obj.model_dump(by_alias=True) diff --git a/vision_agent/tools/tools_types.py b/vision_agent/tools/tools_types.py index aa0e430f..25c2ec93 100644 --- a/vision_agent/tools/tools_types.py +++ b/vision_agent/tools/tools_types.py @@ -28,7 +28,6 @@ class FineTuning(BaseModel): model_config = ConfigDict(populate_by_name=True) job_id: UUID = Field(alias="jobId") - postprocessing: Optional[str] = None @field_serializer("job_id") def serialize_job_id(self, job_id: UUID, _info: SerializationInfo) -> str: @@ -42,6 +41,7 @@ class Florence2FtRequest(BaseModel): task: PromptTask tool: str prompt: Optional[str] = "" + postprocessing: Optional[str] = None fine_tuning: Optional[FineTuning] = Field(None, alias="fineTuning")