dspy icon indicating copy to clipboard operation
dspy copied to clipboard

Load LLMs served on Databricks with DSPY

Open chouksep opened this issue 6 months ago • 1 comments

I am trying to load language model in dspy framework and LLM I am trying to use is the "databricks-mixtral-8x7b-instruct" served on databricks as endpoint. How I can integrate dspy with databricks language model? I looked into modules and there is a module for databricks (https://github.com/stanfordnlp/dspy/blob/main/dsp/modules/databricks.py ) but when trying to load class "Databricks", it shows dspy has no attribute "Databricks". I wanted to know how I can resolve it.

chouksep avatar Feb 28 '24 21:02 chouksep

Hi @chouksep , what version of dspy are you using? I believe some recent updates to the library may have fixed it. Feel free to try again and see if that resolves this!

arnavsinghvi11 avatar Mar 01 '24 17:03 arnavsinghvi11