Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -84,9 +84,9 @@ vector_store.populate_vectors(dataset=None)
|
|
| 84 |
# model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 85 |
|
| 86 |
# load model orca-mini general purpose model
|
| 87 |
-
model_name = '
|
| 88 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 89 |
-
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 90 |
|
| 91 |
# Define the chatbot response function
|
| 92 |
def chatbot_response(user_input):
|
|
|
|
| 84 |
# model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 85 |
|
| 86 |
# load model orca-mini general purpose model
|
| 87 |
+
model_name = 'meta-llama/Meta-Llama-3-8B'
|
| 88 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name,low_cpu_mem_usage=True)
|
| 89 |
+
model = AutoModelForCausalLM.from_pretrained(model_name,low_cpu_mem_usage=True)
|
| 90 |
|
| 91 |
# Define the chatbot response function
|
| 92 |
def chatbot_response(user_input):
|