Spaces:
Runtime error
Runtime error
Update interaction.py
Browse files- interaction.py +2 -1
interaction.py
CHANGED
@@ -9,7 +9,8 @@ from transformers import AutoTokenizer, LlamaForCausalLM
|
|
9 |
from utils import SteamGenerationMixin
|
10 |
|
11 |
auth_token = os.getenv("AUTH_TOKEN")
|
12 |
-
|
|
|
13 |
class MindBot(object):
|
14 |
def __init__(self, model_path, tokenizer_path,if_int8=False):
|
15 |
# self.device = torch.device("cuda")
|
|
|
9 |
from utils import SteamGenerationMixin
|
10 |
|
11 |
auth_token = os.getenv("AUTH_TOKEN")
|
12 |
+
print('^_^ auth_token:',os.getenv("AUTH_TOKEN"),'!!!!!!!!!!')
|
13 |
+
print('^_^:secret_token',os.getenv("SECRET_TOKEN"),'!!!!!!!!!!')
|
14 |
class MindBot(object):
|
15 |
def __init__(self, model_path, tokenizer_path,if_int8=False):
|
16 |
# self.device = torch.device("cuda")
|