├── .gitignore ├── LICENSE ├── MCSD ├── README.md ├── evaluation.py ├── inference │ ├── __init__.py │ ├── generate.py │ └── strategies.py └── model │ ├── __init__.py │ └── llama_tree_attn │ ├── __init__.py │ ├── configuration_llama.py │ ├── convert_llama_weights_to_hf.py │ ├── modeling_llama.py │ ├── tokenization_llama.py │ └── tokenization_llama_fast.py ├── README.md └── dataset └── wmt_ende.json /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/LICENSE -------------------------------------------------------------------------------- /MCSD/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/README.md -------------------------------------------------------------------------------- /MCSD/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/evaluation.py -------------------------------------------------------------------------------- /MCSD/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MCSD/inference/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/inference/generate.py -------------------------------------------------------------------------------- /MCSD/inference/strategies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/inference/strategies.py -------------------------------------------------------------------------------- /MCSD/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MCSD/model/llama_tree_attn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/model/llama_tree_attn/__init__.py -------------------------------------------------------------------------------- /MCSD/model/llama_tree_attn/configuration_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/model/llama_tree_attn/configuration_llama.py -------------------------------------------------------------------------------- /MCSD/model/llama_tree_attn/convert_llama_weights_to_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/model/llama_tree_attn/convert_llama_weights_to_hf.py -------------------------------------------------------------------------------- /MCSD/model/llama_tree_attn/modeling_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/model/llama_tree_attn/modeling_llama.py -------------------------------------------------------------------------------- /MCSD/model/llama_tree_attn/tokenization_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/model/llama_tree_attn/tokenization_llama.py -------------------------------------------------------------------------------- /MCSD/model/llama_tree_attn/tokenization_llama_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/MCSD/model/llama_tree_attn/tokenization_llama_fast.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/README.md -------------------------------------------------------------------------------- /dataset/wmt_ende.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NJUNLP/MCSD/HEAD/dataset/wmt_ende.json --------------------------------------------------------------------------------