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