Spaces:

universalsoftware
/

uchat

Sleeping

shoom013 commited on Sep 15, 2024

Commit

631eefb

verified ·

1 Parent(s): 94bbce1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ from llama_index.llms.huggingface import (
 system_propmpt = "You are a friendly Chatbot."
 # "facebook/blenderbot-400M-distill",  facebook/blenderbot-400M-distill , BAAI/bge-small-en-v1.5
-Settings.llm = HuggingFaceLLM(model_name="HuggingFaceH4/zephyr-7b-gemma-sft-v0.1",
                               device_map="auto",
                               system_prompt = system_propmpt,
                               context_window=4096,

 system_propmpt = "You are a friendly Chatbot."
 # "facebook/blenderbot-400M-distill",  facebook/blenderbot-400M-distill , BAAI/bge-small-en-v1.5
+Settings.llm = HuggingFaceLLM(model_name="stabilityai/stablelm-zephyr-3b",
                               device_map="auto",
                               system_prompt = system_propmpt,
                               context_window=4096,