Spaces:
Runtime error
Runtime error
re-fixed the gate
Browse files
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 |
-
|
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
|