跳至主内容

Xinference[Xorbits 推理]

https://inference.readthedocs.io/en/latest/index.html

API Base, Key

# env variable
os.environ['XINFERENCE_API_BASE'] = "http://127.0.0.1:9997/v1"
os.environ['XINFERENCE_API_KEY'] = "anything" #[optional] no api key required

示例用法 - Embedding

from litellm import embedding
import os

os.environ['XINFERENCE_API_BASE'] = "http://127.0.0.1:9997/v1"
response = embedding(
model="xinference/bge-base-en",
input=["good morning from litellm"],
)
print(response)

示例用法 api_base 参数

from litellm import embedding
import os

response = embedding(
model="xinference/bge-base-en",
api_base="http://127.0.0.1:9997/v1",
input=["good morning from litellm"],
)
print(response)

支持的模型

此处列出的所有模型 https://inference.readthedocs.io/en/latest/models/builtin/embedding/index.html 均受支持

模型名称函数调用
bge-base-enembedding(model="xinference/bge-base-en", input)
bge-base-en-v1.5embedding(model="xinference/bge-base-en-v1.5", input)
bge-base-zhembedding(model="xinference/bge-base-zh", input)
bge-base-zh-v1.5embedding(model="xinference/bge-base-zh-v1.5", input)
bge-large-enembedding(model="xinference/bge-large-en", input)
bge-large-en-v1.5embedding(model="xinference/bge-large-en-v1.5", input)
bge-large-zhembedding(model="xinference/bge-large-zh", input)
bge-large-zh-noinstructembedding(model="xinference/bge-large-zh-noinstruct", input)
bge-large-zh-v1.5embedding(model="xinference/bge-large-zh-v1.5", input)
bge-small-en-v1.5embedding(model="xinference/bge-small-en-v1.5", input)
bge-small-zhembedding(model="xinference/bge-small-zh", input)
bge-small-zh-v1.5embedding(model="xinference/bge-small-zh-v1.5", input)
e5-large-v2embedding(model="xinference/e5-large-v2", input)
gte-baseembedding(model="xinference/gte-base", input)
gte-largeembedding(model="xinference/gte-large", input)
jina-embeddings-v2-base-enembedding(model="xinference/jina-embeddings-v2-base-en", input)
jina-embeddings-v2-small-enembedding(model="xinference/jina-embeddings-v2-small-en", input)
multilingual-e5-largeembedding(model="xinference/multilingual-e5-large", input)