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 |
|