LICENSE
README.md
pyproject.toml
setup.py
llm_foundry.egg-info/PKG-INFO
llm_foundry.egg-info/SOURCES.txt
llm_foundry.egg-info/dependency_links.txt
llm_foundry.egg-info/requires.txt
llm_foundry.egg-info/top_level.txt
llmfoundry/__init__.py
llmfoundry/callbacks/__init__.py
llmfoundry/callbacks/fdiff_callback.py
llmfoundry/callbacks/generate_callback.py
llmfoundry/callbacks/monolithic_ckpt_callback.py
llmfoundry/callbacks/resumption_callbacks.py
llmfoundry/callbacks/scheduled_gc_callback.py
llmfoundry/data/__init__.py
llmfoundry/data/datasets.py
llmfoundry/data/denoising.py
llmfoundry/data/packing.py
llmfoundry/data/text_data.py
llmfoundry/data/finetuning/__init__.py
llmfoundry/data/finetuning/collator.py
llmfoundry/data/finetuning/dataloader.py
llmfoundry/data/finetuning/tasks.py
llmfoundry/models/__init__.py
llmfoundry/models/model_registry.py
llmfoundry/models/hf/__init__.py
llmfoundry/models/hf/hf_causal_lm.py
llmfoundry/models/hf/hf_fsdp.py
llmfoundry/models/hf/hf_prefix_lm.py
llmfoundry/models/hf/hf_t5.py
llmfoundry/models/hf/model_wrapper.py
llmfoundry/models/layers/__init__.py
llmfoundry/models/layers/attention.py
llmfoundry/models/layers/blocks.py
llmfoundry/models/layers/norm.py
llmfoundry/models/mpt/__init__.py
llmfoundry/models/mpt/configuration_mpt.py
llmfoundry/models/mpt/modeling_mpt.py
llmfoundry/models/utils/__init__.py
llmfoundry/models/utils/adapt_tokenizer.py
llmfoundry/models/utils/hf_prefixlm_converter.py
llmfoundry/models/utils/meta_init_context.py
llmfoundry/models/utils/param_init_fns.py
llmfoundry/optim/__init__.py
llmfoundry/optim/adaptive_lion.py
llmfoundry/optim/lion.py
llmfoundry/optim/outlier_detection.py
llmfoundry/utils/__init__.py
llmfoundry/utils/builders.py
llmfoundry/utils/config_utils.py
tests/test_data_prep_scripts.py
tests/test_dataloader.py
tests/test_flash_triton_torch.py
tests/test_hf_v_mpt.py
tests/test_icl_datasets.py
tests/test_init_fn.py
tests/test_model.py
tests/test_onnx.py
tests/test_tokenizer.py
tests/test_training.py