tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-06 19:25:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: update config

Browse Source
This commit is contained in:
arkohut 2024-10-11 14:28:09 +08:00
parent 95f9b4a494
commit b027890c7e
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
memos/config.py
View File

@ -15,16 +15,19 @@ import io
class VLMSettings(BaseModel): class VLMSettings(BaseModel):
enabled: bool = True enabled: bool = True
modelname: str = "moondream" modelname: str = "minicpm-v"
endpoint: str = "http://localhost:11434" endpoint: str = "http://localhost:11434"
token: str = "" token: str = ""
concurrency: int = 1 concurrency: int = 1
force_jpeg: bool = False # some vlm models do not support webp
prompt: str = "请帮描述这个图片中的内容，包括画面格局、出现的视觉元素等" force_jpeg: bool = True
# prompt for vlm to extract caption
prompt: str = "请帮描述这个图片中的内容，包括画面格局、出现的视觉元素等"
class OCRSettings(BaseModel): class OCRSettings(BaseModel):
enabled: bool = True enabled: bool = True
# will by ignored if use_local is True
endpoint: str = "http://localhost:5555/predict" endpoint: str = "http://localhost:5555/predict"
token: str = "" token: str = ""
concurrency: int = 1 concurrency: int = 1
@ -35,13 +38,16 @@ class OCRSettings(BaseModel):
class EmbeddingSettings(BaseModel): class EmbeddingSettings(BaseModel):
enabled: bool = True enabled: bool = True
num_dim: int = 768 num_dim: int = 768
# will be ignored if use_local is True
endpoint: str = "http://localhost:11434/api/embed" endpoint: str = "http://localhost:11434/api/embed"
model: str = "jinaai/jina-embeddings-v2-base-zh" model: str = "jinaai/jina-embeddings-v2-base-zh"
# pull model from huggingface by default, make it true if you want to pull from modelscope
use_modelscope: bool = False use_modelscope: bool = False
use_local: bool = True use_local: bool = True
class TypesenseSettings(BaseModel): class TypesenseSettings(BaseModel):
# is disabled by default, and right now is quite unnecessary
enabled: bool = False enabled: bool = False
host: str = "localhost" host: str = "localhost"
port: str = "8108" port: str = "8108"
@ -65,7 +71,7 @@ class Settings(BaseSettings):
# Server settings # Server settings
server_host: str = "0.0.0.0" server_host: str = "0.0.0.0"
server_port: int = 8080 server_port: int = 8839
# VLM plugin settings # VLM plugin settings
vlm: VLMSettings = VLMSettings() vlm: VLMSettings = VLMSettings()