Spaces:
Sleeping
Sleeping
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)
|