v1.0.1

d99506f3 · chenzk · 61e92904 · d99506f3 · d99506f3 · d99506f3
Commit d99506f3 authored Dec 03, 2024 by chenzk
20 changed files
--- a/examples/doremi/tests/test_doremi_loss.py
+++ b/examples/doremi/tests/test_doremi_loss.py
--- a/examples/doremi/tests/test_doremi_sampler.py
+++ b/examples/doremi/tests/test_doremi_sampler.py
--- a/examples/doremi/tests/test_doremi_utils.py
+++ b/examples/doremi/tests/test_doremi_utils.py
--- a/examples/doremi/tests/utils.py
+++ b/examples/doremi/tests/utils.py
--- a/examples/doremi/train_doremi.py
+++ b/examples/doremi/train_doremi.py
--- a/examples/doremi/train_reference.py
+++ b/examples/doremi/train_reference.py
--- a/examples/doremi/utils.py
+++ b/examples/doremi/utils.py
--- a/examples/llama/README.md
+++ b/examples/llama/README.md
--- a/parallel/tensor_parallel/__init__.py
+++ b/parallel/tensor_parallel/__init__.py
--- a/examples/llama/convert_hf_to_nanotron.py
+++ b/examples/llama/convert_hf_to_nanotron.py
--- a/examples/llama/convert_nanotron_to_hf.py
+++ b/examples/llama/convert_nanotron_to_hf.py
--- a/examples/llama/convert_weights.py
+++ b/examples/llama/convert_weights.py
--- a/examples/llama/requirements.txt
+++ b/examples/llama/requirements.txt
--- a/examples/llama/tests/test_conversion.py
+++ b/examples/llama/tests/test_conversion.py
--- a/examples/llama/tests/test_conversion.py.orig
+++ b/examples/llama/tests/test_conversion.py.orig
--- a/examples/llama/tests/utils.py
+++ b/examples/llama/tests/utils.py
--- a/examples/mamba/README.md
+++ b/examples/mamba/README.md
--- a/examples/mamba/assets/loss_mamba.png
+++ b/examples/mamba/assets/loss_mamba.png
--- a/examples/mamba/config.py
+++ b/examples/mamba/config.py
--- a/examples/mamba/config_mamba.yaml
+++ b/examples/mamba/config_mamba.yaml