Jofthomas HF staff commited on
Commit
d478b6b
1 Parent(s): a3e8ddf
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -116,7 +116,7 @@ class Invoke(BaseModel):
116
 
117
  def generate_text(messages: List[str], npc:str):
118
  print(npc)
119
- if npc in main_npcs:
120
  system_prompt=general_npc_prompt+"/n "+main_npc_system_prompts[npc]
121
  else:
122
  system_prompt="you're a character in a video game. Play along."
 
116
 
117
  def generate_text(messages: List[str], npc:str):
118
  print(npc)
119
+ if npc in main_npc_system_prompts:
120
  system_prompt=general_npc_prompt+"/n "+main_npc_system_prompts[npc]
121
  else:
122
  system_prompt="you're a character in a video game. Play along."