GuoPD commited on
Commit
dcbaaf5
1 Parent(s): b3ca596

modify: update 13B chat model

Browse files
generation_config.json CHANGED
@@ -10,5 +10,6 @@
10
  "top_p": 0.85,
11
  "repetition_penalty": 1.1,
12
  "do_sample": true,
13
- "transformers_version": "4.29.2"
 
14
  }
 
10
  "top_p": 0.85,
11
  "repetition_penalty": 1.1,
12
  "do_sample": true,
13
+ "transformers_version": "4.29.2",
14
+ "no_add_eos": true
15
  }
modeling_baichuan.py CHANGED
@@ -572,9 +572,7 @@ class BaichuanForCausalLM(BaichuanPreTrainedModel):
572
  elif message['role'] == 'assistant':
573
  round_input = [
574
  self.generation_config.assistant_token_id
575
- ] + content_tokens + [
576
- self.generation_config.eos_token_id
577
- ] + round_input
578
  else:
579
  raise ValueError(f"message role not supported yet: {message['role']}")
580
  total_input = total_input[-max_input_tokens:] # truncate left
 
572
  elif message['role'] == 'assistant':
573
  round_input = [
574
  self.generation_config.assistant_token_id
575
+ ] + content_tokens + round_input
 
 
576
  else:
577
  raise ValueError(f"message role not supported yet: {message['role']}")
578
  total_input = total_input[-max_input_tokens:] # truncate left
pytorch_model-00001-of-00003.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc177b37a53447a91c165c0ba53629e5e4eb74ff7c3981185aca18e9556ba59e
3
- size 9972281188
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:493368dd0c14f9e93a9359dcdbbe1640a70c5691851d502867e46df5e3e72e50
3
+ size 9972279780
pytorch_model-00002-of-00003.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9ab48918420e7f7bc965c919e53c57041909a99b430b5472589d6418cc13ee77
3
- size 9947421360
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83b12d88f4aa459f377dd714866fa2b81c99e88fefec727286ea302515ef4aed
3
+ size 9947419824
pytorch_model-00003-of-00003.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d07290b5739ea068dec0f816789596d7cc33c4f4e72cd700fe7cf5084ab05d9
3
- size 6610200289
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:788ef65efe2ce412605cc81a056db2784235cec23cfccd1d9d3bed64b3e4d3be
3
+ size 6610199329