ferrariedhgs commited on
Commit
926d5a4
·
verified ·
1 Parent(s): 77770dc

Upload 2 files

Browse files
Files changed (1) hide show
  1. misc_f.py +1 -1
misc_f.py CHANGED
@@ -347,7 +347,7 @@ def generate_game(llm, max_retries: int = 3) -> dict:
347
  raw = _generate_text(
348
  llm,
349
  messages=[
350
- {"role": "system", "content": GENERATE_GAME_PROMPT},
351
  {"role": "user", "content": '{"task":"generate_room"}'},
352
  ],
353
  max_new_tokens=1024,
 
347
  raw = _generate_text(
348
  llm,
349
  messages=[
350
+ {"role": "system", "content": "/nothink\n\n"+GENERATE_GAME_PROMPT},
351
  {"role": "user", "content": '{"task":"generate_room"}'},
352
  ],
353
  max_new_tokens=1024,