1 ano atrás · 4c9cc7d223
--- a/src/llama_recipes/configs/__init__.py
+++ b/src/llama_recipes/configs/__init__.py
--- a/src/llama_recipes/configs/datasets.py
+++ b/src/llama_recipes/configs/datasets.py
--- a/src/llama_recipes/configs/fsdp.py
+++ b/src/llama_recipes/configs/fsdp.py
--- a/src/llama_recipes/configs/peft.py
+++ b/src/llama_recipes/configs/peft.py
--- a/src/llama_recipes/configs/training.py
+++ b/src/llama_recipes/configs/training.py
--- a/src/llama_recipes/datasets/__init__.py
+++ b/src/llama_recipes/datasets/__init__.py
--- a/src/llama_recipes/datasets/alpaca_dataset.py
+++ b/src/llama_recipes/datasets/alpaca_dataset.py
--- a/src/llama_recipes/datasets/grammar_dataset/__init__.py
+++ b/src/llama_recipes/datasets/grammar_dataset/__init__.py
--- a/src/llama_recipes/datasets/grammar_dataset/grammar_dataset.py
+++ b/src/llama_recipes/datasets/grammar_dataset/grammar_dataset.py
--- a/src/llama_recipes/datasets/grammar_dataset/grammar_dataset_process.ipynb
+++ b/src/llama_recipes/datasets/grammar_dataset/grammar_dataset_process.ipynb
--- a/src/llama_recipes/datasets/samsum_dataset.py
+++ b/src/llama_recipes/datasets/samsum_dataset.py
--- a/src/llama_recipes/datasets/utils.py
+++ b/src/llama_recipes/datasets/utils.py
--- a/src/llama_recipes/finetuning.py
+++ b/src/llama_recipes/finetuning.py
--- a/src/llama_recipes/inference/README.md
+++ b/src/llama_recipes/inference/README.md
--- a/src/llama_recipes/inference/chat_completion.py
+++ b/src/llama_recipes/inference/chat_completion.py
--- a/src/llama_recipes/inference/chat_utils.py
+++ b/src/llama_recipes/inference/chat_utils.py
--- a/src/llama_recipes/inference/chats.json
+++ b/src/llama_recipes/inference/chats.json
--- a/src/llama_recipes/inference/checkpoint_converter_fsdp_hf.py
+++ b/src/llama_recipes/inference/checkpoint_converter_fsdp_hf.py
--- a/src/llama_recipes/inference/code-llama/code_completion_example.py
+++ b/src/llama_recipes/inference/code-llama/code_completion_example.py
--- a/src/llama_recipes/inference/code-llama/code_completion_prompt.txt
+++ b/src/llama_recipes/inference/code-llama/code_completion_prompt.txt
--- a/src/llama_recipes/inference/code-llama/code_infilling_example.py
+++ b/src/llama_recipes/inference/code-llama/code_infilling_example.py
--- a/src/llama_recipes/inference/code-llama/code_infilling_prompt.txt
+++ b/src/llama_recipes/inference/code-llama/code_infilling_prompt.txt
--- a/src/llama_recipes/inference/hf-text-generation-inference/README.md
+++ b/src/llama_recipes/inference/hf-text-generation-inference/README.md
--- a/src/llama_recipes/inference/hf-text-generation-inference/merge_lora_weights.py
+++ b/src/llama_recipes/inference/hf-text-generation-inference/merge_lora_weights.py
--- a/src/llama_recipes/inference/inference.py
+++ b/src/llama_recipes/inference/inference.py
--- a/src/llama_recipes/inference/model_utils.py
+++ b/src/llama_recipes/inference/model_utils.py
--- a/src/llama_recipes/inference/safety_utils.py
+++ b/src/llama_recipes/inference/safety_utils.py
--- a/src/llama_recipes/inference/samsum_prompt.txt
+++ b/src/llama_recipes/inference/samsum_prompt.txt
--- a/src/llama_recipes/inference/vLLM_inference.py
+++ b/src/llama_recipes/inference/vLLM_inference.py
--- a/src/llama_recipes/model_checkpointing/__init__.py
+++ b/src/llama_recipes/model_checkpointing/__init__.py
--- a/src/llama_recipes/model_checkpointing/checkpoint_handler.py
+++ b/src/llama_recipes/model_checkpointing/checkpoint_handler.py
--- a/src/llama_recipes/policies/__init__.py
+++ b/src/llama_recipes/policies/__init__.py
--- a/src/llama_recipes/policies/activation_checkpointing_functions.py
+++ b/src/llama_recipes/policies/activation_checkpointing_functions.py
--- a/src/llama_recipes/policies/anyprecision_optimizer.py
+++ b/src/llama_recipes/policies/anyprecision_optimizer.py
--- a/src/llama_recipes/policies/mixed_precision.py
+++ b/src/llama_recipes/policies/mixed_precision.py
--- a/src/llama_recipes/policies/wrapping.py
+++ b/src/llama_recipes/policies/wrapping.py
--- a/src/llama_recipes/utils/__init__.py
+++ b/src/llama_recipes/utils/__init__.py
--- a/src/llama_recipes/utils/config_utils.py
+++ b/src/llama_recipes/utils/config_utils.py
--- a/src/llama_recipes/utils/dataset_utils.py
+++ b/src/llama_recipes/utils/dataset_utils.py
--- a/src/llama_recipes/utils/fsdp_utils.py
+++ b/src/llama_recipes/utils/fsdp_utils.py
--- a/src/llama_recipes/utils/memory_utils.py
+++ b/src/llama_recipes/utils/memory_utils.py
--- a/src/llama_recipes/utils/train_utils.py
+++ b/src/llama_recipes/utils/train_utils.py