Vito99 commited on
Commit
33ad9ac
1 Parent(s): 496eeae

Update awesome_chat.py

Browse files
Files changed (1) hide show
  1. awesome_chat.py +1 -1
awesome_chat.py CHANGED
@@ -71,7 +71,7 @@ use_completion = config["use_completion"]
71
 
72
  # consistent: wrong msra model name
73
  LLM_encoding = LLM
74
- if LLM == "text-davinci-003":
75
  LLM_encoding = "text-davinci-003"
76
  task_parsing_highlight_ids = get_token_ids_for_task_parsing(LLM_encoding)
77
  choose_model_highlight_ids = get_token_ids_for_choose_model(LLM_encoding)
 
71
 
72
  # consistent: wrong msra model name
73
  LLM_encoding = LLM
74
+ if LLM == "gpt-3.5-turbo-0301":
75
  LLM_encoding = "text-davinci-003"
76
  task_parsing_highlight_ids = get_token_ids_for_task_parsing(LLM_encoding)
77
  choose_model_highlight_ids = get_token_ids_for_choose_model(LLM_encoding)