Crystalcareai commited on
Commit
912f59b
·
verified ·
1 Parent(s): 1423a0e

Update modeling_quiet.py

Browse files
Files changed (1) hide show
  1. modeling_quiet.py +2 -5
modeling_quiet.py CHANGED
@@ -1531,11 +1531,8 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
1531
  if self.tokenizer is not None:
1532
  self.start_token_id = self.tokenizer.convert_tokens_to_ids("<|startthought|>")
1533
  else:
1534
- self.start_token_id = 0
1535
- if self.start_token_id == 0:
1536
- self.start_token_id = self.tokenizer.bos_token_id
1537
- self.tokenizer_has_start_thought_token = False
1538
- elif self.use_start_thought_token:
1539
  # base_start_id = self.tokenizer.convert_tokens_to_ids(self.initial_start_token)
1540
  base_start_id = self.tokenizer.encode(self.initial_start_token, add_special_tokens=False)[0]
1541
  if self.initialize_thought_embedding_to_normal:
 
1531
  if self.tokenizer is not None:
1532
  self.start_token_id = self.tokenizer.convert_tokens_to_ids("<|startthought|>")
1533
  else:
1534
+ self.start_token_id = self.tokenizer.bos_token_id if self.tokenizer else 0
1535
+ self.tokenizer_has_start_thought_token = False if self.tokenizer else True
 
 
 
1536
  # base_start_id = self.tokenizer.convert_tokens_to_ids(self.initial_start_token)
1537
  base_start_id = self.tokenizer.encode(self.initial_start_token, add_special_tokens=False)[0]
1538
  if self.initialize_thought_embedding_to_normal: