Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -106,7 +106,7 @@ import logging
|
|
106 |
logging.basicConfig(level=logging.INFO)
|
107 |
|
108 |
# Initialize the Llama model with chat format set to "llama-2"
|
109 |
-
llm = Llama(model_path="
|
110 |
|
111 |
# Define the system prompt
|
112 |
system_prompt = (
|
|
|
106 |
logging.basicConfig(level=logging.INFO)
|
107 |
|
108 |
# Initialize the Llama model with chat format set to "llama-2"
|
109 |
+
llm = Llama(model_path="./law-chat.Q2_K.gguf", chat_format="llama-2")
|
110 |
|
111 |
# Define the system prompt
|
112 |
system_prompt = (
|