__init__.py
|
feat(moe): add moe module (#182)
|
2023-09-27 15:54:53 +08:00 |
linear.py
|
refactor code
|
2023-10-09 20:23:32 +08:00 |
loss.py
|
initial commit
|
2023-07-06 12:55:23 +08:00 |
modeling_internlm.py
|
support evaluation with fstp
|
2023-10-09 18:01:06 +08:00 |
modeling_moe.py
|
feat(moe): add moe module (#182)
|
2023-09-27 15:54:53 +08:00 |
moe.py
|
feat(moe): add moe module (#182)
|
2023-09-27 15:54:53 +08:00 |
multi_head_attention.py
|
support fstp and refactor code
|
2023-10-09 17:26:20 +08:00 |
norm.py
|
Merge develop to main (#233)
|
2023-08-24 22:03:04 +08:00 |
utils.py
|
refactor code
|
2023-10-09 20:23:32 +08:00 |