SynLayers commited on
Commit
43bd07d
·
verified ·
1 Parent(s): 00c8491

Upload demo/infer/run_caption_bbox_infer.py with huggingface_hub

Browse files
demo/infer/run_caption_bbox_infer.py CHANGED
@@ -31,12 +31,11 @@ def resolve_default_bbox_model() -> str:
31
  candidates = [
32
  PROJECT_ROOT if (PROJECT_ROOT / "config.json").exists() and (PROJECT_ROOT / "tokenizer_config.json").exists() else None,
33
  PROJECT_ROOT / "Bbox-caption-8b",
34
- Path("/project/llmsvgen/share/data/kmw_layered_checkpoint/Bbox-caption-8b"),
35
  ]
36
  for candidate in candidates:
37
  if candidate and candidate.exists():
38
  return str(candidate)
39
- return str(Path("/project/llmsvgen/share/data/kmw_layered_checkpoint/Bbox-caption-8b"))
40
 
41
 
42
  CAPTION_BBOX_PROMPT_TOP_LEFT = (
 
31
  candidates = [
32
  PROJECT_ROOT if (PROJECT_ROOT / "config.json").exists() and (PROJECT_ROOT / "tokenizer_config.json").exists() else None,
33
  PROJECT_ROOT / "Bbox-caption-8b",
 
34
  ]
35
  for candidate in candidates:
36
  if candidate and candidate.exists():
37
  return str(candidate)
38
+ return "SynLayers/Bbox-caption-8b"
39
 
40
 
41
  CAPTION_BBOX_PROMPT_TOP_LEFT = (