v1.0.8

dfcb88ff · chenzk · dfcb88ff · dfcb88ff · dfcb88ff · dfcb88ff
Commit dfcb88ff authored Dec 04, 2024 by chenzk
20 changed files
--- a/src/nanotron/distributed.py
+++ b/src/nanotron/distributed.py
--- a/src/nanotron/fp8/__init__.py
+++ b/src/nanotron/fp8/__init__.py
--- a/src/nanotron/fp8/constants.py
+++ b/src/nanotron/fp8/constants.py
--- a/src/nanotron/fp8/dtypes.py
+++ b/src/nanotron/fp8/dtypes.py
--- a/src/nanotron/fp8/kernel.py
+++ b/src/nanotron/fp8/kernel.py
--- a/src/nanotron/fp8/linear.py
+++ b/src/nanotron/fp8/linear.py
--- a/src/nanotron/fp8/meta.py
+++ b/src/nanotron/fp8/meta.py
--- a/src/nanotron/fp8/parameter.py
+++ b/src/nanotron/fp8/parameter.py
--- a/src/nanotron/fp8/tensor.py
+++ b/src/nanotron/fp8/tensor.py
--- a/src/nanotron/fp8/utils.py
+++ b/src/nanotron/fp8/utils.py
--- a/src/nanotron/generation/__init__.py
+++ b/src/nanotron/generation/__init__.py
--- a/src/nanotron/generation/decode.py
+++ b/src/nanotron/generation/decode.py
--- a/src/nanotron/generation/generate_store.py
+++ b/src/nanotron/generation/generate_store.py
--- a/src/nanotron/generation/sampler.py
+++ b/src/nanotron/generation/sampler.py
--- a/src/nanotron/helpers.py
+++ b/src/nanotron/helpers.py
--- a/src/nanotron/logging.py
+++ b/src/nanotron/logging.py
--- a/src/nanotron/models/__init__.py
+++ b/src/nanotron/models/__init__.py
--- a/src/nanotron/models/base.py
+++ b/src/nanotron/models/base.py
--- a/src/nanotron/models/llama.py
+++ b/src/nanotron/models/llama.py
--- a/src/nanotron/models/starcoder2.py
+++ b/src/nanotron/models/starcoder2.py