Ankitajadhav commited on
Commit
88ca944
·
verified ·
1 Parent(s): b7e9232

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ vector_store.populate_vectors(dataset=None)
82
  # model = AutoModelForCausalLM.from_pretrained(model_name)
83
 
84
  # load model orca-mini general purpose model
85
- model_name = 'meta-llama/Meta-Llama-3-8B'
86
  tokenizer = AutoTokenizer.from_pretrained(model_name,low_cpu_mem_usage=True)
87
  model = AutoModelForCausalLM.from_pretrained(model_name,low_cpu_mem_usage=True)
88
 
 
82
  # model = AutoModelForCausalLM.from_pretrained(model_name)
83
 
84
  # load model orca-mini general purpose model
85
+ model_name = 'google/gemma-2-9b'
86
  tokenizer = AutoTokenizer.from_pretrained(model_name,low_cpu_mem_usage=True)
87
  model = AutoModelForCausalLM.from_pretrained(model_name,low_cpu_mem_usage=True)
88