mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-06 02:48:22 +00:00
feat(embedding): update dimensions parameter to support optional type
This commit is contained in:
parent
ef7c4daa9a
commit
681fb22170
1 changed files with 1 additions and 1 deletions
|
|
@ -22,7 +22,7 @@ class BaseEmbeddingModel(ABC):
|
|||
def __init__(
|
||||
self,
|
||||
model_name: str = "",
|
||||
dimensions: int = 1024,
|
||||
dimensions: int | None = 1024,
|
||||
max_batch_size: int = 10,
|
||||
max_retries: int = 3,
|
||||
raise_exception: bool = True,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue