mirror of
https://github.com/tcsenpai/pensieve.git
synced 2025-06-06 19:25:24 +00:00
feat(vlm): update prompt
This commit is contained in:
parent
31838ac4a3
commit
b39d651b0c
@ -10,19 +10,9 @@ import uvicorn
|
|||||||
import os
|
import os
|
||||||
import io
|
import io
|
||||||
|
|
||||||
from unittest.mock import patch
|
|
||||||
from transformers.dynamic_module_utils import get_imports
|
|
||||||
|
|
||||||
def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
|
|
||||||
if not str(filename).endswith("modeling_florence2.py"):
|
|
||||||
return get_imports(filename)
|
|
||||||
imports = get_imports(filename)
|
|
||||||
imports.remove("flash_attn")
|
|
||||||
return imports
|
|
||||||
|
|
||||||
|
|
||||||
PLUGIN_NAME = "vlm"
|
PLUGIN_NAME = "vlm"
|
||||||
PROMPT = "描述这张图片的内容"
|
PROMPT = "请帮我尽量详尽的描述这个图片中的内容,包括文字内容、视觉元素等"
|
||||||
|
|
||||||
router = APIRouter(tags=[PLUGIN_NAME], responses={404: {"description": "Not found"}})
|
router = APIRouter(tags=[PLUGIN_NAME], responses={404: {"description": "Not found"}})
|
||||||
|
|
||||||
@ -33,8 +23,6 @@ concurrency = None
|
|||||||
semaphore = None
|
semaphore = None
|
||||||
force_jpeg = None
|
force_jpeg = None
|
||||||
use_local = None
|
use_local = None
|
||||||
florence_model = None
|
|
||||||
florence_processor = None
|
|
||||||
torch_dtype = None
|
torch_dtype = None
|
||||||
|
|
||||||
# Configure logger
|
# Configure logger
|
||||||
|
Loading…
x
Reference in New Issue
Block a user