v1.0.3

71e79847 · chenzk · 71e79847 · 71e79847 · 71e79847 · 71e79847
Commit 71e79847 authored Dec 03, 2024 by chenzk
20 changed files
--- a/examples/doremi/tests/test_doremi_dataloader.py
+++ b/examples/doremi/tests/test_doremi_dataloader.py
--- 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/examples/llama/__init__.py
+++ b/examples/llama/__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
+transformers==4.39.3
--- 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