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

trying keras

Browse files
Files changed (1) hide show
  1. backend.py +2 -2
backend.py CHANGED
@@ -20,9 +20,9 @@ import keras_nlp
20
  #from llama_index import LangchainEmbedding, ServiceContext
21
 
22
  # Set the backbend before importing Keras
23
- os.environ["KERAS_BACKEND"] = "jax"
24
  # Avoid memory fragmentation on JAX backend.
25
- os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "1.00"
26
 
27
 
28
  #os.getenv("KAGGLE_USERNAME")
 
20
  #from llama_index import LangchainEmbedding, ServiceContext
21
 
22
  # Set the backbend before importing Keras
23
+ #os.environ["KERAS_BACKEND"] = "jax"
24
  # Avoid memory fragmentation on JAX backend.
25
+ #os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "1.00"
26
 
27
 
28
  #os.getenv("KAGGLE_USERNAME")