Spaces:
Sleeping
Sleeping
orionweller
commited on
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
|