timxiaohangt commited on
Commit
81326fa
·
1 Parent(s): 5a0fec8
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ os.environ['CUDA_VISIBLE_DEVICES'] = "0"
12
  MODEL_NAME = "RegularizedSelfPlay/sppo_forward1reverse5-0.1-PromptABC-Mistral-7B-Instruct-SPPO-Iter3" # Example: "meta-llama/Llama-2-7b-chat-hf"
13
 
14
  # Load model and tokenizer
15
- tokenizer = LlamaTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
16
  llm = LLM(
17
  model=MODEL_NAME,
18
  # revision="1296dc8fd9b21e6424c9c305c06db9ae60c03ace",
 
12
  MODEL_NAME = "RegularizedSelfPlay/sppo_forward1reverse5-0.1-PromptABC-Mistral-7B-Instruct-SPPO-Iter3" # Example: "meta-llama/Llama-2-7b-chat-hf"
13
 
14
  # Load model and tokenizer
15
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
16
  llm = LLM(
17
  model=MODEL_NAME,
18
  # revision="1296dc8fd9b21e6424c9c305c06db9ae60c03ace",