Omnibus commited on
Commit
9734f46
·
verified ·
1 Parent(s): 5753e75

Update app.py

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