gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add embeding compute task

Browse Source
This commit is contained in:
tsukasa
2023-08-31 15:45:02 +08:00
parent d1b8795eed
commit 071452fe19
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/compute_task.py
+1
View File
@@ -63,6 +63,7 @@ class ComputeTask:
self.params["model_name"] = model_name self.params["model_name"] = model_name
else: else:
self.params["model_name"] = "text-embedding-ada-002" self.params["model_name"] = "text-embedding-ada-002"
self.params["input"] = input self.params["input"] = input
def display(self) -> str: def display(self) -> str: