Spaces:
Sleeping
Sleeping
Commit
·
3e7db0b
1
Parent(s):
f448409
working cpu
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ logger = logging.getLogger(__name__)
|
|
| 30 |
login(token=os.environ['HF_TOKEN'])
|
| 31 |
|
| 32 |
# Global variables
|
| 33 |
-
CUR_MODEL = "
|
| 34 |
BASE_MODEL = "meta-llama/Llama-2-7b-hf"
|
| 35 |
tokenizer = None
|
| 36 |
model = None
|
|
|
|
| 30 |
login(token=os.environ['HF_TOKEN'])
|
| 31 |
|
| 32 |
# Global variables
|
| 33 |
+
CUR_MODEL = "Samaya-AI/Promptriever-Llama2-v1"
|
| 34 |
BASE_MODEL = "meta-llama/Llama-2-7b-hf"
|
| 35 |
tokenizer = None
|
| 36 |
model = None
|