YWMditto 28635755f5 [fix bug] Fix the error that RotaryEmbedding is converted to a non-fp32 format during training, and add a compatible method for the llama model. (#239) ... Co-authored-by: YWMditto <862779238@qq.com>		2023-08-26 17:48:08 +08:00
..
__init__.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
embedding.py	[fix bug] Fix the error that RotaryEmbedding is converted to a non-fp32 format during training, and add a compatible method for the llama model. (#239)	2023-08-26 17:48:08 +08:00
linear.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
loss.py	initial commit	2023-07-06 12:55:23 +08:00
metrics.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
modeling_internlm.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
multi_head_attention.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
norm.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
utils.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00