Spaces:
Runtime error
Runtime error
valentin urena
commited on
Update chess_board.py
Browse files- chess_board.py +1 -1
chess_board.py
CHANGED
|
@@ -18,7 +18,7 @@ class Game:
|
|
| 18 |
self.model_id = 'kaggle://valentinbaltazar/gemma-chess/keras/gemma_2b_en_chess'
|
| 19 |
self.sampler = keras_nlp.samplers.TopKSampler(k=50, temperature=0.7)
|
| 20 |
self.model = keras_nlp.models.GemmaCausalLM.from_preset(self.model_id)
|
| 21 |
-
self.
|
| 22 |
|
| 23 |
def compile_model(self):
|
| 24 |
self.model.compile(sampler=self.sampler)
|
|
|
|
| 18 |
self.model_id = 'kaggle://valentinbaltazar/gemma-chess/keras/gemma_2b_en_chess'
|
| 19 |
self.sampler = keras_nlp.samplers.TopKSampler(k=50, temperature=0.7)
|
| 20 |
self.model = keras_nlp.models.GemmaCausalLM.from_preset(self.model_id)
|
| 21 |
+
self.compile_model()
|
| 22 |
|
| 23 |
def compile_model(self):
|
| 24 |
self.model.compile(sampler=self.sampler)
|