v1.0.1

d99506f3 · chenzk · 61e92904 · d99506f3 · d99506f3 · d99506f3
Commit d99506f3 authored Dec 03, 2024 by chenzk
20 changed files
--- a/config/utils_config.py
+++ b/config/utils_config.py
--- a/constants.py
+++ b/constants.py
--- a/data/collator.py
+++ b/data/collator.py
--- a/data/dataloader_builder.py
+++ b/data/dataloader_builder.py
--- a/data/nanoset.py
+++ b/data/nanoset.py
--- a/data/utils.py
+++ b/data/utils.py
--- a/dataloader.py
+++ b/dataloader.py
@@ -119,7 +119,6 @@ def get_datasets(
                hf_dataset_or_datasets,
                hf_dataset_config_name,
                split=split,
-                cache_dir=".",
            )
    else:
        raise ValueError(f"hf_dataset_or_datasets must be a dict or string but is {type(hf_dataset_or_datasets)}")

--- a/distributed.py
+++ b/distributed.py
--- a/fp8/__init__.py
+++ b/fp8/__init__.py
--- a/fp8/constants.py
+++ b/fp8/constants.py
--- a/fp8/dtypes.py
+++ b/fp8/dtypes.py
--- a/fp8/kernel.py
+++ b/fp8/kernel.py
--- a/fp8/linear.py
+++ b/fp8/linear.py
--- a/fp8/meta.py
+++ b/fp8/meta.py
--- a/fp8/parameter.py
+++ b/fp8/parameter.py
--- a/fp8/tensor.py
+++ b/fp8/tensor.py
--- a/fp8/utils.py
+++ b/fp8/utils.py
--- a/generation/__init__.py
+++ b/generation/__init__.py
--- a/generation/decode.py
+++ b/generation/decode.py
--- a/generation/generate_store.py
+++ b/generation/generate_store.py