LICENSE
MANIFEST.in
README.md
requirements.txt
setup.py
SwissArmyTransformer/__init__.py
SwissArmyTransformer/arguments.py
SwissArmyTransformer/transformer_defaults.py
SwissArmyTransformer.egg-info/PKG-INFO
SwissArmyTransformer.egg-info/SOURCES.txt
SwissArmyTransformer.egg-info/dependency_links.txt
SwissArmyTransformer.egg-info/requires.txt
SwissArmyTransformer.egg-info/top_level.txt
SwissArmyTransformer/data_utils/__init__.py
SwissArmyTransformer/data_utils/configure_data.py
SwissArmyTransformer/data_utils/datasets.py
SwissArmyTransformer/data_utils/hf_dataset.py
SwissArmyTransformer/data_utils/samplers.py
SwissArmyTransformer/generation/__init__.py
SwissArmyTransformer/generation/autoregressive_sampling.py
SwissArmyTransformer/generation/cuda2d_sampling.py
SwissArmyTransformer/generation/magnify.py
SwissArmyTransformer/generation/utils.py
SwissArmyTransformer/generation/sampling_strategies/__init__.py
SwissArmyTransformer/generation/sampling_strategies/base_strategy.py
SwissArmyTransformer/generation/sampling_strategies/beam_search_strategy.py
SwissArmyTransformer/generation/sampling_strategies/iterative_entfilter_strategy.py
SwissArmyTransformer/model/__init__.py
SwissArmyTransformer/model/base_model.py
SwissArmyTransformer/model/cached_autoregressive_model.py
SwissArmyTransformer/model/encoder_decoder_model.py
SwissArmyTransformer/model/mixins.py
SwissArmyTransformer/model/transformer.py
SwissArmyTransformer/model/finetune/__init__.py
SwissArmyTransformer/model/finetune/adapter.py
SwissArmyTransformer/model/finetune/ffadd.py
SwissArmyTransformer/model/finetune/lora.py
SwissArmyTransformer/model/finetune/mlp_head.py
SwissArmyTransformer/model/finetune/prompt_tuning.py
SwissArmyTransformer/model/official/__init__.py
SwissArmyTransformer/model/official/bert_model.py
SwissArmyTransformer/model/official/cait_model.py
SwissArmyTransformer/model/official/clip_model.py
SwissArmyTransformer/model/official/cuda2d_model.py
SwissArmyTransformer/model/official/distill_model.py
SwissArmyTransformer/model/official/glm130B_model.py
SwissArmyTransformer/model/official/glm_model.py
SwissArmyTransformer/model/official/mae_model.py
SwissArmyTransformer/model/official/roberta_model.py
SwissArmyTransformer/model/official/t5_model.py
SwissArmyTransformer/model/official/vit_model.py
SwissArmyTransformer/model/official/yolos_model.py
SwissArmyTransformer/model/position_embedding/__init__.py
SwissArmyTransformer/model/position_embedding/rotary_embeddings.py
SwissArmyTransformer/model/position_embedding/sincos2d.py
SwissArmyTransformer/mpu/__init__.py
SwissArmyTransformer/mpu/cross_entropy.py
SwissArmyTransformer/mpu/data.py
SwissArmyTransformer/mpu/initialize.py
SwissArmyTransformer/mpu/layers.py
SwissArmyTransformer/mpu/mappings.py
SwissArmyTransformer/mpu/utils.py
SwissArmyTransformer/ops/__init__.py
SwissArmyTransformer/ops/layernorm.py
SwissArmyTransformer/ops/local_attention_function.py
SwissArmyTransformer/resources/__init__.py
SwissArmyTransformer/resources/download.py
SwissArmyTransformer/resources/urls.py
SwissArmyTransformer/tokenization/__init__.py
SwissArmyTransformer/tokenization/hf_tokenizer.py
SwissArmyTransformer/tokenization/cogview/__init__.py
SwissArmyTransformer/tokenization/cogview/sp_tokenizer.py
SwissArmyTransformer/tokenization/cogview/templates.py
SwissArmyTransformer/tokenization/cogview/unified_tokenizer.py
SwissArmyTransformer/tokenization/cogview/vqvae_tokenizer.py
SwissArmyTransformer/tokenization/cogview/vqvae/__init__.py
SwissArmyTransformer/tokenization/cogview/vqvae/api.py
SwissArmyTransformer/tokenization/cogview/vqvae/vqvae_diffusion.py
SwissArmyTransformer/tokenization/cogview/vqvae/vqvae_zc.py
SwissArmyTransformer/tokenization/embed_assets/chinese_sentencepiece/cog-pretrain.model
SwissArmyTransformer/tokenization/embed_assets/chinese_sentencepiece/cog-pretrain.vocab
SwissArmyTransformer/tokenization/embed_assets/english_tokenizer/bert-base-uncased-vocab.txt
SwissArmyTransformer/tokenization/embed_assets/english_tokenizer/bert-large-uncased-vocab.txt
SwissArmyTransformer/tokenization/embed_assets/english_tokenizer/gpt2-merges.txt
SwissArmyTransformer/tokenization/embed_assets/english_tokenizer/gpt2-vocab.json
SwissArmyTransformer/tokenization/embed_assets/english_tokenizer/roberta-merges.txt
SwissArmyTransformer/tokenization/embed_assets/english_tokenizer/roberta-vocab.json
SwissArmyTransformer/tokenization/glm/__init__.py
SwissArmyTransformer/tokenization/glm/sp_tokenizer.py
SwissArmyTransformer/tokenization/glm/tokenization.py
SwissArmyTransformer/tokenization/glm/tokenization_gpt2.py
SwissArmyTransformer/tokenization/icetk_glm_130B/__init__.py
SwissArmyTransformer/tokenization/icetk_glm_130B/ice_tokenizer.py
SwissArmyTransformer/tokenization/icetk_glm_130B/tokenizer.py
SwissArmyTransformer/training/__init__.py
SwissArmyTransformer/training/deepspeed_training.py
SwissArmyTransformer/training/deepspeed_zero0.json
SwissArmyTransformer/training/deepspeed_zero1.json
SwissArmyTransformer/training/deepspeed_zero2.json
SwissArmyTransformer/training/learning_rates.py
SwissArmyTransformer/training/model_io.py
SwissArmyTransformer/training/utils.py