Th3BossC commited on
Commit
bc76c08
1 Parent(s): d1e5b6e

forgot to uncomment important stuff

Browse files
Files changed (1) hide show
  1. ChitChat/common/utils.py +2 -3
ChitChat/common/utils.py CHANGED
@@ -13,8 +13,8 @@ small_model = AutoModelForCausalLM.from_pretrained(small_model_name)
13
  small_tokenizer = AutoTokenizer.from_pretrained(default_model)
14
  small_tokenizer.pad_token = small_tokenizer.eos_token
15
 
16
- # large_model = T5ForConditionalGeneration.from_pretrained(large_model_name)
17
- # large_tokenizer = T5Tokenizer.from_pretrained(large_model_name)
18
 
19
  def getChatHistory(user):
20
  if user.history is None:
@@ -57,7 +57,6 @@ def conversation(user, userInput):
57
  return small_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens = True)
58
 
59
  def complexChat(userInput):
60
- return "hi"
61
  input_ids = large_tokenizer(userInput, return_tensors="pt").input_ids
62
  outputs = large_model.generate(input_ids,
63
  min_length = 10,
 
13
  small_tokenizer = AutoTokenizer.from_pretrained(default_model)
14
  small_tokenizer.pad_token = small_tokenizer.eos_token
15
 
16
+ large_model = T5ForConditionalGeneration.from_pretrained(large_model_name)
17
+ large_tokenizer = T5Tokenizer.from_pretrained(large_model_name)
18
 
19
  def getChatHistory(user):
20
  if user.history is None:
 
57
  return small_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens = True)
58
 
59
  def complexChat(userInput):
 
60
  input_ids = large_tokenizer(userInput, return_tensors="pt").input_ids
61
  outputs = large_model.generate(input_ids,
62
  min_length = 10,