Omnibus commited on
Commit
07e39b8
1 Parent(s): 960e1dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ def generate(prompt, history,max_new_tokens,health,seed,temperature=temperature,
78
  print(f'cnt:: {cnt}')
79
  if cnt > MAX_HISTORY:
80
  history1 = compress_history(str(history), temperature, top_p, repetition_penalty)
81
- formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,stats=stats,dice=random.randint(1,10))}, {prompt}", history)
82
  stream = client.text_generation(prompt=formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
83
  output = ""
84
 
 
78
  print(f'cnt:: {cnt}')
79
  if cnt > MAX_HISTORY:
80
  history1 = compress_history(str(history), temperature, top_p, repetition_penalty)
81
+ formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,stats=stats,dice=random.randint(1,10))}, USER INPUT:\n{prompt}", history)
82
  stream = client.text_generation(prompt=formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
83
  output = ""
84