huckiyang commited on
Commit
3a7f01f
·
1 Parent(s): cb62b20

re-fixed the gate

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,9 +5,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
5
  from trl import AutoModelForCausalLMWithValueHead
6
  from safetensors.torch import load_file
7
  import logging
 
8
 
9
  # Set up logging
10
- logging.basicConfig(level=logging.INFO)
11
 
12
  # Constants
13
  THRESHOLD = 2 # From Plan2Align
 
5
  from trl import AutoModelForCausalLMWithValueHead
6
  from safetensors.torch import load_file
7
  import logging
8
+ from huggingface_hub import login
9
 
10
  # Set up logging
11
+ login(token=os.environ.get("LA_NAME"))
12
 
13
  # Constants
14
  THRESHOLD = 2 # From Plan2Align