gufett0 commited on
Commit
d312028
·
1 Parent(s): bbe77cb

trying keras

Browse files
Files changed (2) hide show
  1. backend.py +2 -2
  2. interface.py +1 -1
backend.py CHANGED
@@ -25,8 +25,8 @@ os.environ["KERAS_BACKEND"] = "jax"
25
  os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "1.00"
26
 
27
 
28
- os.getenv("KAGGLE_USERNAME")
29
- os.getenv["KAGGLE_KEY"]
30
 
31
  """huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
32
  login(huggingface_token)
 
25
  os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "1.00"
26
 
27
 
28
+ #os.getenv("KAGGLE_USERNAME")
29
+ #os.getenv["KAGGLE_KEY"]
30
 
31
  """huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
32
  login(huggingface_token)
interface.py CHANGED
@@ -69,7 +69,7 @@ import keras_nlp
69
  if not streamed_response:
70
  yield CompletionResponse(text="No response generated.", delta="No response generated.")"""
71
 
72
-
73
  class GemmaLLMInterface(CustomLLM):
74
  model: keras_nlp.models.GemmaCausalLM = None
75
  context_window: int = 8192
 
69
  if not streamed_response:
70
  yield CompletionResponse(text="No response generated.", delta="No response generated.")"""
71
 
72
+ # for Keras
73
  class GemmaLLMInterface(CustomLLM):
74
  model: keras_nlp.models.GemmaCausalLM = None
75
  context_window: int = 8192