Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = '
|
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 |
|