Merge pull request #107 from tpoisonooo/patch-1

fix(lightrag_siliconcloud_demo.py): max_token_size
This commit is contained in:
zrguo
2024-10-23 11:32:59 +08:00
committed by GitHub

View File

@@ -30,7 +30,7 @@ async def embedding_func(texts: list[str]) -> np.ndarray:
texts, texts,
model="netease-youdao/bce-embedding-base_v1", model="netease-youdao/bce-embedding-base_v1",
api_key=os.getenv("UPSTAGE_API_KEY"), api_key=os.getenv("UPSTAGE_API_KEY"),
max_token_size=int(512 * 1.5) max_token_size=512
) )