Langchain

Install

mamba install langchain openai transformers xformers

from langchain_openai import OpenAI

llm = OpenAI(temperature=0.9)

text = "What would be a good company name for a company that makes colorful socks?"

print(llm.invoke(text))

from langchain import HuggingFacePipeline

llm = HuggingFacePipeline.from_model_id(

model_id="databricks/dolly-v2-12b",

task="text-generation",

model_kwargs={"temperature": 0, "max_length": 64},

)

text = "Explain to me the difference between nuclear fission and fusion."

print(llm(text))

from langchain.llms.bedrock import Bedrock