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

chore: change default concurrency

Browse Source
This commit is contained in:
arkohut 2024-11-06 16:55:22 +08:00
parent f524622728
commit c1ffa007ff
3 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
memos/config.py
View File

@ -11,7 +11,6 @@ from pydantic_settings import (
from pydantic import BaseModel, SecretStr from pydantic import BaseModel, SecretStr
import yaml import yaml
from collections import OrderedDict from collections import OrderedDict
import io
import typer import typer
@ -19,7 +18,7 @@ class VLMSettings(BaseModel):
modelname: str = "minicpm-v" modelname: str = "minicpm-v"
endpoint: str = "http://localhost:11434" endpoint: str = "http://localhost:11434"
token: str = "" token: str = ""
concurrency: int = 1 concurrency: int = 8
# some vlm models do not support webp # some vlm models do not support webp
force_jpeg: bool = True force_jpeg: bool = True
# prompt for vlm to extract caption # prompt for vlm to extract caption
@ -30,7 +29,7 @@ class OCRSettings(BaseModel):
# will by ignored if use_local is 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 = 8
use_local: bool = True use_local: bool = True
force_jpeg: bool = False force_jpeg: bool = False

4
memos/default_config.yaml
View File

@ -14,7 +14,7 @@ default_plugins:
# using ollama as the vlm server # using ollama as the vlm server
vlm: vlm:
concurrency: 1 concurrency: 8
endpoint: http://localhost:11434 endpoint: http://localhost:11434
force_jpeg: true force_jpeg: true
modelname: minicpm-v modelname: minicpm-v
@ -23,7 +23,7 @@ vlm:
# using local ocr # using local ocr
ocr: ocr:
concurrency: 1 concurrency: 8
# this is not used if use_local is true # this is not used if use_local is true
endpoint: http://localhost:5555/predict endpoint: http://localhost:5555/predict
force_jpeg: false force_jpeg: false

1
memos/server.py
View File

@ -17,7 +17,6 @@ import json
import cv2 import cv2
from PIL import Image from PIL import Image
from secrets import compare_digest from secrets import compare_digest
import functools
import logging import logging
from .config import get_database_path, settings from .config import get_database_path, settings