Ankitajadhav commited on
Commit
2daa438
·
verified ·
1 Parent(s): abab33a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = 'mistral-7b-openorca.gguf2.Q4_0.gguf'
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):