apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-18 13:55:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

tuning code

Browse source
This commit is contained in:
geekan 2024-05-18 15:23:21 +08:00
parent 6d906ce1ed
commit e03db313a2
9 changed files with 40 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/rag/factories/embedding.py
View file

@ -89,9 +89,9 @@ class RAGEmbeddingFactory(GenericFactory):
return OllamaEmbedding(**params)
def _try_set_model_and_batch_size(self, params: dict):
"""Set the model_name and embed_batch_size only when they are specified."""
"""Set the model and embed_batch_size only when they are specified."""
if config.embedding.model:
params["model_name"] = config.embedding.model
params["model"] = config.embedding.model
if config.embedding.embed_batch_size:
params["embed_batch_size"] = config.embedding.embed_batch_size