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

fix: use modelscope for sub config

Browse Source
This commit is contained in:
arkohut 2024-09-10 01:59:37 +08:00
parent 513b41e300
commit 6bbf1a8a68
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
memos/config.py
View File

@ -20,6 +20,7 @@ class VLMSettings(BaseModel):
concurrency: int = 1 concurrency: int = 1
force_jpeg: bool = False force_jpeg: bool = False
use_local: bool = True use_local: bool = True
use_modelscope: bool = False
class OCRSettings(BaseModel): class OCRSettings(BaseModel):
@ -37,6 +38,7 @@ class EmbeddingSettings(BaseModel):
num_dim: int = 768 num_dim: int = 768
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"
use_modelscope: bool = False
class Settings(BaseSettings): class Settings(BaseSettings):
@ -51,8 +53,6 @@ class Settings(BaseSettings):
default_library: str = "screenshots" default_library: str = "screenshots"
screenshots_dir: str = os.path.join(base_dir, "screenshots") screenshots_dir: str = os.path.join(base_dir, "screenshots")
use_modelscope: bool = False
typesense_host: str = "localhost" typesense_host: str = "localhost"
typesense_port: str = "8108" typesense_port: str = "8108"
typesense_protocol: str = "http" typesense_protocol: str = "http"