hf context window
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def main():
|
|
51 |
if provider == 'huggingface':
|
52 |
llm_name = st.text_input(
|
53 |
"Enter LLM namespace/model-name",
|
54 |
-
value="
|
55 |
)
|
56 |
|
57 |
# Also give the user the option for different embedding models, too
|
@@ -119,7 +119,7 @@ def main():
|
|
119 |
Settings.embed_model = HuggingFaceInferenceAPIEmbedding(
|
120 |
model_name=embed_name
|
121 |
)
|
122 |
-
|
123 |
else:
|
124 |
raise NotImplementedError(f"{provider} is not supported yet")
|
125 |
|
|
|
51 |
if provider == 'huggingface':
|
52 |
llm_name = st.text_input(
|
53 |
"Enter LLM namespace/model-name",
|
54 |
+
value="HuggingFaceH4/zephyr-7b-alpha",
|
55 |
)
|
56 |
|
57 |
# Also give the user the option for different embedding models, too
|
|
|
119 |
Settings.embed_model = HuggingFaceInferenceAPIEmbedding(
|
120 |
model_name=embed_name
|
121 |
)
|
122 |
+
Settings.context_window = 4096
|
123 |
else:
|
124 |
raise NotImplementedError(f"{provider} is not supported yet")
|
125 |
|