Skip dimensions for local embedding model
This commit is contained in:
parent
be44c243fd
commit
c5dbabf558
@ -64,7 +64,7 @@ class GlobalModelManager:
|
|||||||
"model": self.external_model_name,
|
"model": self.external_model_name,
|
||||||
"input": texts,
|
"input": texts,
|
||||||
}
|
}
|
||||||
if self.external_dimensions and self.external_model_name != "text-embedding-ada-002":
|
if self.external_dimensions and self.external_model_name not in ("text-embedding-ada-002", "local-embedding"):
|
||||||
payload["dimensions"] = self.external_dimensions
|
payload["dimensions"] = self.external_dimensions
|
||||||
|
|
||||||
response = requests.post(
|
response = requests.post(
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user