Spaces:
Paused
Paused
Update app 18 apr.py
Browse files- app 18 apr.py +1 -1
app 18 apr.py
CHANGED
|
@@ -10,7 +10,7 @@ from typing import Optional
|
|
| 10 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 11 |
from sentence_transformers import SentenceTransformer
|
| 12 |
|
| 13 |
-
# ββ Configuration ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 14 |
CHECKPOINT = "selfrag/selfrag_llama2_7b"
|
| 15 |
K_PASSAGES = 5
|
| 16 |
MAX_NEW_TOKS = 300
|
|
|
|
| 10 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 11 |
from sentence_transformers import SentenceTransformer
|
| 12 |
|
| 13 |
+
# ββ Configuration. ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 14 |
CHECKPOINT = "selfrag/selfrag_llama2_7b"
|
| 15 |
K_PASSAGES = 5
|
| 16 |
MAX_NEW_TOKS = 300
|