Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def set_custom_prompt():
|
|
| 46 |
return prompt
|
| 47 |
|
| 48 |
# Load LLM model
|
| 49 |
-
llm_model = CTransformers(
|
| 50 |
model_type = 'llama',
|
| 51 |
max_new_token = 512,
|
| 52 |
temperature=0.5)
|
|
|
|
| 46 |
return prompt
|
| 47 |
|
| 48 |
# Load LLM model
|
| 49 |
+
llm_model = CTransformers(model='TheBloke/Llama-2-7B-Chat-GGML',
|
| 50 |
model_type = 'llama',
|
| 51 |
max_new_token = 512,
|
| 52 |
temperature=0.5)
|