done

landing-ai · Sep 3, 2024 · b479ce3 · b479ce3
1 parent 59698df
commit b479ce3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py
@@ -61,6 +61,8 @@ def register_tool(imports: Optional[List] = None) -> Callable:
  def decorator(tool: Callable) -> Callable:
  import inspect
 
+ print("try")
+
  from .tools import get_tool_descriptions, get_tools_df, get_tools_info
 
  global TOOLS, TOOLS_DF, TOOL_DESCRIPTIONS, TOOL_DOCSTRING, TOOLS_INFO

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -1639,7 +1639,10 @@ def florence2_phrase_grounding_fine_tune(
  if status is not JobStatus.SUCCEEDED:
  raise FineTuneModelIsNotReady()
 
- task = PromptTask[task]
+ task = PromptTask[
+ "CAPTION_TO_PHRASE_GROUNDING"
+ ] # hardcode to <CAPTION_TO_PHRASE_GROUNDING> for now
+
  if task is PromptTask.OBJECT_DETECTION:
  prompt = ""