LICENSE
MANIFEST.in
README.md
pyproject.toml
firelink/__init__.py
firelink/_recipe_registry.py
firelink/recipe_interfaces.py
firelink/_cli/__init__.py
firelink/_cli/cp.py
firelink/_cli/download.py
firelink/_cli/ls.py
firelink/_cli/run.py
firelink/_cli/subcommand.py
firelink/_cli/tune.py
firelink/_cli/validate.py
firelink/config/__init__.py
firelink/config/_errors.py
firelink/config/_instantiate.py
firelink/config/_parse.py
firelink/config/_utils.py
firelink/config/_validate.py
firelink/data/__init__.py
firelink/data/_chat_formats.py
firelink/data/_common.py
firelink/data/_converters.py
firelink/data/_instruct_templates.py
firelink/data/_types.py
firelink/data/_utils.py
firelink/datasets/__init__.py
firelink/datasets/_alpaca.py
firelink/datasets/_chat.py
firelink/datasets/_cnn_dailymail.py
firelink/datasets/_concat.py
firelink/datasets/_grammar.py
firelink/datasets/_instruct.py
firelink/datasets/_packed.py
firelink/datasets/_preference.py
firelink/datasets/_samsum.py
firelink/datasets/_slimorca.py
firelink/datasets/_stack_exchanged_paired.py
firelink/datasets/_text_completion.py
firelink/datasets/_wikitext.py
firelink/models/__init__.py
firelink/models/convert_weights.py
firelink/models/clip/__init__.py
firelink/models/clip/_component_builders.py
firelink/models/clip/_model_builders.py
firelink/models/clip/_position_embeddings.py
firelink/models/clip/_transforms.py
firelink/models/code_llama2/__init__.py
firelink/models/code_llama2/_model_builders.py
firelink/models/gemma/__init__.py
firelink/models/gemma/_component_builders.py
firelink/models/gemma/_model_builders.py
firelink/models/gemma/_tokenizer.py
firelink/models/gemma/rms_norm.py
firelink/models/gemma/transformer.py
firelink/models/llama2/__init__.py
firelink/models/llama2/_component_builders.py
firelink/models/llama2/_model_builders.py
firelink/models/llama2/_model_utils.py
firelink/models/llama2/_tokenizer.py
firelink/models/llama3/__init__.py
firelink/models/llama3/_component_builders.py
firelink/models/llama3/_model_builders.py
firelink/models/llama3/_model_utils.py
firelink/models/llama3/_tokenizer.py
firelink/models/llama3_1/__init__.py
firelink/models/llama3_1/_component_builders.py
firelink/models/llama3_1/_model_builders.py
firelink/models/llama3_1/_position_embeddings.py
firelink/models/mistral/__init__.py
firelink/models/mistral/_component_builders.py
firelink/models/mistral/_model_builders.py
firelink/models/mistral/_tokenizer.py
firelink/models/phi3/__init__.py
firelink/models/phi3/_component_builders.py
firelink/models/phi3/_convert_weights.py
firelink/models/phi3/_model_builders.py
firelink/models/phi3/_position_embeddings.py
firelink/models/phi3/_tokenizer.py
firelink/models/qwen2/__init__.py
firelink/models/qwen2/_component_builders.py
firelink/models/qwen2/_convert_weights.py
firelink/models/qwen2/_model_builders.py
firelink/models/qwen2/_positional_embeddings.py
firelink/models/qwen2/_tokenizer.py
firelink/modules/__init__.py
firelink/modules/attention.py
firelink/modules/common_utils.py
firelink/modules/feed_forward.py
firelink/modules/kv_cache.py
firelink/modules/layer_norm.py
firelink/modules/lr_schedulers.py
firelink/modules/position_embeddings.py
firelink/modules/rms_norm.py
firelink/modules/transformer.py
firelink/modules/vision_transformer.py
firelink/modules/loss/__init__.py
firelink/modules/loss/dpo.py
firelink/modules/low_precision/__init__.py
firelink/modules/low_precision/_register_nf4_dispatch_ops.py
firelink/modules/low_precision/_utils.py
firelink/modules/low_precision/nf4_linear.py
firelink/modules/peft/__init__.py
firelink/modules/peft/lora.py
firelink/modules/peft/peft_utils.py
firelink/modules/rlhf/__init__.py
firelink/modules/rlhf/utils/__init__.py
firelink/modules/rlhf/utils/_convert_weights.py
firelink/modules/tokenizers/__init__.py
firelink/modules/tokenizers/_sentencepiece.py
firelink/modules/tokenizers/_tiktoken.py
firelink/modules/tokenizers/_utils.py
firelink/modules/transforms/__init__.py
firelink/modules/transforms/_transforms.py
firelink/modules/transforms/vision_utils/__init__.py
firelink/modules/transforms/vision_utils/get_canvas_best_fit.py
firelink/modules/transforms/vision_utils/resize_with_pad.py
firelink/modules/transforms/vision_utils/tile_crop.py
firelink/utils/__init__.py
firelink/utils/_device.py
firelink/utils/_distributed.py
firelink/utils/_generation.py
firelink/utils/_profiler.py
firelink/utils/_version.py
firelink/utils/activations.py
firelink/utils/argparse.py
firelink/utils/collate.py
firelink/utils/constants.py
firelink/utils/logging.py
firelink/utils/memory.py
firelink/utils/metric_logging.py
firelink/utils/pooling.py
firelink/utils/precision.py
firelink/utils/quantization.py
firelink/utils/seed.py
firelink/utils/_checkpointing/__init__.py
firelink/utils/_checkpointing/_checkpointer.py
firelink/utils/_checkpointing/_checkpointer_utils.py
firelink_shrine.egg-info/PKG-INFO
firelink_shrine.egg-info/SOURCES.txt
firelink_shrine.egg-info/dependency_links.txt
firelink_shrine.egg-info/entry_points.txt
firelink_shrine.egg-info/requires.txt
firelink_shrine.egg-info/top_level.txt
recipes/__init__.py
recipes/eleuther_eval.py
recipes/full_finetune_distributed.py
recipes/full_finetune_single_device.py
recipes/generate.py
recipes/lora_dpo_distributed.py
recipes/lora_dpo_single_device.py
recipes/lora_finetune_distributed.py
recipes/lora_finetune_single_device.py
recipes/qat_distributed.py
recipes/quantize.py
recipes/configs/eleuther_evaluation.yaml
recipes/configs/generation.yaml
recipes/configs/quantization.yaml
recipes/configs/code_llama2/7B_full_low_memory.yaml
recipes/configs/code_llama2/7B_lora_single_device.yaml
recipes/configs/code_llama2/7B_qlora_single_device.yaml
recipes/configs/dev/8B_full_experimental.yaml
recipes/configs/dev/llama2/13B_lora_fsdp2.yaml
recipes/configs/dev/llama2/70B_lora_fsdp2.yaml
recipes/configs/dev/llama2/70B_qlora_fsdp2.yaml
recipes/configs/dev/llama2/7B_lora_fsdp2.yaml
recipes/configs/dev/llama2/7B_qlora_fsdp2.yaml
recipes/configs/gemma/2B_full.yaml
recipes/configs/gemma/2B_lora.yaml
recipes/configs/gemma/2B_lora_single_device.yaml
recipes/configs/gemma/2B_qlora_single_device.yaml
recipes/configs/gemma/7B_full.yaml
recipes/configs/gemma/7B_lora.yaml
recipes/configs/gemma/7B_lora_single_device.yaml
recipes/configs/gemma/7B_qlora_single_device.yaml
recipes/configs/llama2/13B_full.yaml
recipes/configs/llama2/13B_lora.yaml
recipes/configs/llama2/13B_qlora_single_device.yaml
recipes/configs/llama2/70B_lora.yaml
recipes/configs/llama2/7B_full.yaml
recipes/configs/llama2/7B_full_low_memory.yaml
recipes/configs/llama2/7B_lora.yaml
recipes/configs/llama2/7B_lora_dpo.yaml
recipes/configs/llama2/7B_lora_dpo_single_device.yaml
recipes/configs/llama2/7B_lora_single_device.yaml
recipes/configs/llama2/7B_qat_full.yaml
recipes/configs/llama2/7B_qlora_single_device.yaml
recipes/configs/llama3/70B_full.yaml
recipes/configs/llama3/70B_lora.yaml
recipes/configs/llama3/8B_full.yaml
recipes/configs/llama3/8B_full_single_device.yaml
recipes/configs/llama3/8B_lora.yaml
recipes/configs/llama3/8B_lora_single_device.yaml
recipes/configs/llama3/8B_qat_full.yaml
recipes/configs/llama3/8B_qlora_single_device.yaml
recipes/configs/llama3_1/70B_full.yaml
recipes/configs/llama3_1/70B_lora.yaml
recipes/configs/llama3_1/8B_full.yaml
recipes/configs/llama3_1/8B_full_single_device.yaml
recipes/configs/llama3_1/8B_lora.yaml
recipes/configs/llama3_1/8B_lora_single_device.yaml
recipes/configs/llama3_1/8B_qlora_single_device.yaml
recipes/configs/mistral/7B_full.yaml
recipes/configs/mistral/7B_full_low_memory.yaml
recipes/configs/mistral/7B_lora.yaml
recipes/configs/mistral/7B_lora_single_device.yaml
recipes/configs/mistral/7B_qlora_single_device.yaml
recipes/configs/phi3/mini_full.yaml
recipes/configs/phi3/mini_full_low_memory.yaml
recipes/configs/phi3/mini_lora.yaml
recipes/configs/phi3/mini_lora_single_device.yaml
recipes/configs/phi3/mini_qlora_single_device.yaml
recipes/configs/qwen2/7B_full.yaml
recipes/configs/qwen2/7B_full_low_memory.yaml
recipes/configs/qwen2/7B_lora.yaml
recipes/configs/qwen2/7B_lora_single_device.yaml
recipes/dev/lora_finetune_fsdp2.py