Update modeling_quiet.py
Browse files- 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 |
-
|
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:
|