navid72m commited on
Commit
8a5f60b
·
1 Parent(s): d2ec37c
Files changed (1) hide show
  1. selection.py +1 -1
selection.py CHANGED
@@ -25,7 +25,7 @@ def find_most_relevant_context(contexts, question):
25
  os.environ["KERAS_BACKEND"] = "jax" # Or "torch" or "tensorflow".
26
  # Avoid memory fragmentation on JAX backend.
27
  os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
28
- os.environ["HF_TOKEN"] = os.getenv['my_repo_token']
29
  # Example usage:
30
 
31
  dataset = load_dataset('json', data_files='train.json',field='data')
 
25
  os.environ["KERAS_BACKEND"] = "jax" # Or "torch" or "tensorflow".
26
  # Avoid memory fragmentation on JAX backend.
27
  os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
28
+ os.environ["HF_TOKEN"] = os.getenv('my_repo_token')
29
  # Example usage:
30
 
31
  dataset = load_dataset('json', data_files='train.json',field='data')