from openai import OpenAI
client = OpenAI(
base_url = "https://integrate.api.nvidia.com/v1",
api_key = "$NVIDIA_API_KEY"
)
completion = client.chat.completions.create(
model="sarvamai/sarvam-m",
messages=[{"role":"user","content":""}],
temperature=0.5,
top_p=1,
max_tokens=16384,
stream=False
)
print(completion.choices[0].message.content)Deploy this model now on your endpoint provider of choice