Commit Graph

15 Commits

Author SHA1 Message Date
duzx16 8127ab6abf Support batch training 2023-03-29 21:15:30 +08:00
duzx16 fbda1206cb Merge branch 'main' into dev_pt
# Conflicts:
#	modeling_chatglm.py
2023-03-29 20:37:39 +08:00
duzx16 812f43f9ff Add p-tuning v2 2023-03-29 20:22:57 +08:00
duzx16 096f3de6b4 Fix context length in get_position_ids 2023-03-28 17:37:46 +08:00
duzx16 4a9b711e61 Close CPU fusion on Mac 2023-03-23 22:43:06 +08:00
duzx16 d2bbc82a2c Fix Chinese punctuation 2023-03-22 14:37:21 +08:00
duzx16 2460dc2430 Remove hardcode bos_token_id 2023-03-19 14:56:15 +08:00
duzx16 42095d42ff Add support for streaming output 2023-03-19 14:31:26 +08:00
duzx16 220f772e9a Fix overflow in FP16 2023-03-16 09:26:05 +08:00
duzx16 f9f74fda55 Set is_parallelizable to False 2023-03-16 00:30:43 +08:00
duzx16 c3dece3f01 Add logit processor for NaN or Inf scores 2023-03-15 18:14:34 +08:00
duzx16 9d1509a1ad Fix default history argument 2023-03-14 18:38:49 +08:00
duzx16 d4832e8142 Add support for float32 2023-03-14 14:49:14 +08:00
duzx16 cd8041ea53 Fix past_key_values 2023-03-14 02:08:43 +08:00
Sengxian d11c6aaed8 Add chatglm-6b 2023-03-14 00:21:01 +08:00