adjust postprocessing

landing-ai · Sep 30, 2024 · 13e415d · 13e415d
1 parent 1dade30
commit 13e415d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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")