File size: 401 Bytes
68dd607
 
 
4e34758
 
68dd607
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from llama_index.llms.huggingface import HuggingFaceLLM
from llama_index.core.llms import ChatMessage
llm = HuggingFaceLLM(
    model_name="internlm/internlm2_5-1_8b-chat",
    tokenizer_name="internlm/internlm2_5-1_8b-chat",
    model_kwargs={"trust_remote_code":True},
    tokenizer_kwargs={"trust_remote_code":True}
)

rsp = llm.chat(messages=[ChatMessage(content="xtuner是什么?")])
print(rsp)