-
- Downloads
Added autoregressive heads and sampling
Showing
- modules/generative_head/composite_head_A.py 68 additions, 13 deletionsmodules/generative_head/composite_head_A.py
- modules/generative_head/convolution_head_A.py 49 additions, 1 deletionmodules/generative_head/convolution_head_A.py
- modules/generative_head/double_substitution_head.py 118 additions, 1 deletionmodules/generative_head/double_substitution_head.py
- modules/generative_head/head_factory.py 3 additions, 1 deletionmodules/generative_head/head_factory.py
- modules/generative_head/substitution_head.py 96 additions, 2 deletionsmodules/generative_head/substitution_head.py
- modules/utils/__init__.py 2 additions, 0 deletionsmodules/utils/__init__.py
- modules/utils/block_convolution.py 36 additions, 0 deletionsmodules/utils/block_convolution.py
- sample/token_generator/basic_generator.py 63 additions, 0 deletionssample/token_generator/basic_generator.py
- sample/token_generator/composite_generator.py 45 additions, 4 deletionssample/token_generator/composite_generator.py
- sample/token_generator/double_substitution_generator.py 81 additions, 1 deletionsample/token_generator/double_substitution_generator.py
- sample/token_generator/substitution_generator.py 66 additions, 0 deletionssample/token_generator/substitution_generator.py
- sample/token_generator/token_generator_factory.py 4 additions, 1 deletionsample/token_generator/token_generator_factory.py
Loading
Please register or sign in to comment