Skip to content

GitLab

Commit 9a4434f5 authored by Rick Ho's avatar Rick Ho
Browse files

add fastmoe.megatron to install list

parent 2bceb76b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 9a4434f5
...@@ -20,7 +20,7 @@ if __name__ == '__main__': ...@@ -20,7 +20,7 @@ if __name__ == '__main__':
author_email='hja20@mails.tsinghua.edu.cn', author_email='hja20@mails.tsinghua.edu.cn',
license='Apache-2', license='Apache-2',
url='https://github.com/laekov/fastmoe', url='https://github.com/laekov/fastmoe',
packages=['fmoe'], packages=['fmoe', 'fmoe.megatron'],
ext_modules=[ ext_modules=[
CUDAExtension( CUDAExtension(
name='fmoe_cuda', name='fmoe_cuda',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment