valentin-ub commited on
Commit
cd064af
·
verified ·
1 Parent(s): 4bc948f

Update chess_board.py

Browse files
Files changed (1) hide show
  1. chess_board.py +2 -2
chess_board.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
  import os
3
  os.environ["KERAS_BACKEND"] = "torch" # "jax", "torch" or "tensorflow"
4
 
@@ -28,7 +28,7 @@ class Game:
28
  def compile_model(self):
29
  self.model.compile(sampler=self.sampler)
30
 
31
- @spaces.GPU
32
  def inference_gemma(self, prompt, max_length=256):
33
  """Inference requires GPU"""
34
  response = self.model.generate(prompt, max_length)
 
1
+ # import spaces
2
  import os
3
  os.environ["KERAS_BACKEND"] = "torch" # "jax", "torch" or "tensorflow"
4
 
 
28
  def compile_model(self):
29
  self.model.compile(sampler=self.sampler)
30
 
31
+ # @spaces.GPU
32
  def inference_gemma(self, prompt, max_length=256):
33
  """Inference requires GPU"""
34
  response = self.model.generate(prompt, max_length)