Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -19,8 +19,10 @@ data = dataset["train"]
|
|
19 |
data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
|
20 |
|
21 |
|
22 |
-
model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
|
23 |
# model_id = "google/gemma-2-2b-it"
|
|
|
|
|
24 |
|
25 |
# use quantization to lower GPU usage
|
26 |
bnb_config = BitsAndBytesConfig(
|
|
|
19 |
data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
|
20 |
|
21 |
|
22 |
+
# model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
|
23 |
# model_id = "google/gemma-2-2b-it"
|
24 |
+
model_id = "meta-llama/Llama-3.2-1B"
|
25 |
+
|
26 |
|
27 |
# use quantization to lower GPU usage
|
28 |
bnb_config = BitsAndBytesConfig(
|