Zimix commited on
Commit
4cc0861
1 Parent(s): 9599983

Update interaction.py

Browse files
Files changed (1) hide show
  1. 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")