valentin urena commited on
Commit
f510d64
·
verified ·
1 Parent(s): 0f0db04

Update chess_board.py

Browse files
Files changed (1) hide show
  1. chess_board.py +1 -1
chess_board.py CHANGED
@@ -27,7 +27,7 @@ class Game:
27
  instruction=f"Predict the next chess move in the sequence {str(self.sequence)}",
28
  response="",)
29
 
30
- output = self.model.generate(prompt, max_length=max_output_len)
31
 
32
  gemma_move = output.split(' ')[-1].strip("'")
33
 
 
27
  instruction=f"Predict the next chess move in the sequence {str(self.sequence)}",
28
  response="",)
29
 
30
+ output = self.model.generate(prompt, max_length=256)
31
 
32
  gemma_move = output.split(' ')[-1].strip("'")
33