orionweller commited on
Commit
3e7db0b
1 Parent(s): f448409

working cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "orionweller/repllama-instruct-hard-positives-v2-joint"
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