Skip to content

GitLab

"megatron/training/arguments.py" did not exist on "b6e0377ba99e3334f94cd38d2bd706e98e227883"
Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 163 Bytes
Newer Older
Jared Casper's avatar
Clean up licensing.  
Jared Casper committed
1 
# Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
mshoeybi's avatar
tested and working  
mshoeybi committed
2
3
4
5 


from .api import (
    generate,
rprenger's avatar
Possibly working beam search  
rprenger committed
6
7 
    generate_and_post_process,
    beam_search_and_post_process)