From dc3e91a97484463dafe68328e8f45e38238d79c3 Mon Sep 17 00:00:00 2001 From: Michelle Casbon Date: Tue, 23 Jan 2024 08:18:41 -0800 Subject: [PATCH] Change get_vocabularies references to point to the grain package. PiperOrigin-RevId: 600794873 --- docs/conf.py | 2 +- docs/conf_sphinx_patch.py | 2 +- setup.py | 2 +- t5x/__init__.py | 2 +- t5x/adafactor.py | 2 +- t5x/adafactor_test.py | 2 +- t5x/assert_gc_disabled_during_import_test_util.py | 2 +- t5x/binary_search.py | 2 +- t5x/binary_search_test.py | 2 +- t5x/checkpoint_importer.py | 2 +- t5x/checkpoint_importer_test.py | 2 +- t5x/checkpoint_utils.py | 2 +- t5x/checkpoint_utils_test.py | 2 +- t5x/checkpoints.py | 2 +- t5x/config_utils.py | 2 +- t5x/configs/__init__.py | 2 +- t5x/configs/runs/__init__.py | 2 +- t5x/contrib/__init__.py | 2 +- t5x/contrib/calm/__init__.py | 2 +- t5x/contrib/calm/decoding.py | 2 +- t5x/contrib/calm/models.py | 2 +- t5x/contrib/moe/__init__.py | 2 +- t5x/contrib/moe/adafactor_utils.py | 2 +- t5x/contrib/moe/checkpoints.py | 2 +- t5x/contrib/moe/checkpoints_test.py | 2 +- t5x/contrib/moe/configs/__init__.py | 2 +- t5x/contrib/moe/configs/runs/__init__.py | 2 +- t5x/contrib/moe/models.py | 2 +- t5x/contrib/moe/models_test.py | 2 +- t5x/contrib/moe/partitioning.py | 2 +- t5x/contrib/moe/partitioning_test.py | 2 +- t5x/contrib/moe/trainer.py | 2 +- t5x/contrib/moe/trainer_test.py | 2 +- t5x/contrib/moe/training_utils.py | 2 +- t5x/contrib/moe/training_utils_test.py | 2 +- t5x/decoding.py | 2 +- t5x/decoding_test.py | 2 +- t5x/disable_gc_during_import.py | 2 +- t5x/disable_gc_during_import_test.py | 2 +- t5x/eval.py | 2 +- t5x/examples/__init__.py | 2 +- t5x/examples/decoder_only/layers.py | 2 +- t5x/examples/decoder_only/layers_test.py | 2 +- t5x/examples/decoder_only/network.py | 2 +- t5x/examples/decoder_only/network_test.py | 2 +- t5x/examples/scalable_t5/__init__.py | 2 +- t5x/examples/scalable_t5/layers.py | 2 +- t5x/examples/scalable_t5/layers_test.py | 2 +- t5x/examples/scalable_t5/mt5/__init__.py | 2 +- t5x/examples/scalable_t5/network.py | 2 +- t5x/examples/scalable_t5/network_test.py | 2 +- t5x/examples/scalable_t5/t5_1_1/__init__.py | 2 +- t5x/examples/scalable_t5/t5_1_1/examples/__init__.py | 2 +- t5x/examples/scalable_t5/umt5/__init__.py | 2 +- t5x/examples/scalable_t5/umt5/architectures/__init__.py | 2 +- t5x/examples/scalable_t5/umt5/optimizer/__init__.py | 2 +- t5x/examples/scalable_t5/umt5/runs/__init__.py | 2 +- t5x/examples/t5/__init__.py | 2 +- t5x/examples/t5/byt5/__init__.py | 2 +- t5x/examples/t5/layers.py | 2 +- t5x/examples/t5/layers_test.py | 2 +- t5x/examples/t5/mt5/__init__.py | 2 +- t5x/examples/t5/network.py | 2 +- t5x/examples/t5/network_test.py | 2 +- t5x/examples/t5/t5_1_0/__init__.py | 2 +- t5x/examples/t5/t5_1_1/__init__.py | 2 +- t5x/examples/t5/t5_1_1/examples/__init__.py | 2 +- t5x/export.py | 2 +- t5x/export_lib.py | 2 +- t5x/fiddle_configs/__init__.py | 2 +- t5x/fiddle_configs/configs/__init__.py | 2 +- t5x/fiddle_configs/configs/finetune.py | 2 +- t5x/fiddle_configs/configs/pretrain.py | 2 +- t5x/fiddle_configs/examples/__init__.py | 2 +- t5x/fiddle_configs/examples/t5_1_1.py | 2 +- t5x/fiddle_configs/examples/t5_1_1_test.py | 2 +- t5x/fiddle_configs/models/__init__.py | 2 +- t5x/fiddle_configs/models/t5_1_1.py | 2 +- t5x/gin_utils.py | 2 +- t5x/gin_utils_test.py | 2 +- t5x/infer.py | 2 +- t5x/interactive_model.py | 2 +- t5x/losses.py | 2 +- t5x/losses_test.py | 2 +- t5x/main.py | 2 +- t5x/metrics.py | 2 +- t5x/metrics_test.py | 2 +- t5x/models.py | 2 +- t5x/models_test.py | 2 +- t5x/optimizers.py | 2 +- t5x/optimizers_test.py | 2 +- t5x/partitioning.py | 2 +- t5x/partitioning_test.py | 2 +- t5x/precompile.py | 2 +- t5x/scripts/__init__.py | 2 +- t5x/scripts/convert_tf_checkpoint.py | 2 +- t5x/scripts/xm_launch.py | 2 +- t5x/state_utils.py | 2 +- t5x/state_utils_test.py | 2 +- t5x/test_utils.py | 2 +- t5x/train.py | 2 +- t5x/train_state.py | 2 +- t5x/train_state_test.py | 2 +- t5x/trainer.py | 2 +- t5x/trainer_test.py | 2 +- t5x/utils.py | 5 +++-- t5x/utils_test.py | 2 +- t5x/version.py | 2 +- 108 files changed, 110 insertions(+), 109 deletions(-) diff --git a/docs/conf.py b/docs/conf.py index d4219e4d2..2f5378272 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/docs/conf_sphinx_patch.py b/docs/conf_sphinx_patch.py index 59eaff051..3210e3e35 100644 --- a/docs/conf_sphinx_patch.py +++ b/docs/conf_sphinx_patch.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/setup.py b/setup.py index 28f89b36d..aa2a4e6bb 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/__init__.py b/t5x/__init__.py index bd301c369..70746335a 100644 --- a/t5x/__init__.py +++ b/t5x/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/adafactor.py b/t5x/adafactor.py index 8fd1f38a5..1920fd5ce 100644 --- a/t5x/adafactor.py +++ b/t5x/adafactor.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/adafactor_test.py b/t5x/adafactor_test.py index 54ead0012..df8f07e38 100644 --- a/t5x/adafactor_test.py +++ b/t5x/adafactor_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/assert_gc_disabled_during_import_test_util.py b/t5x/assert_gc_disabled_during_import_test_util.py index dd2517743..a395327ef 100644 --- a/t5x/assert_gc_disabled_during_import_test_util.py +++ b/t5x/assert_gc_disabled_during_import_test_util.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/binary_search.py b/t5x/binary_search.py index 68ee8f969..36e264c31 100644 --- a/t5x/binary_search.py +++ b/t5x/binary_search.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/binary_search_test.py b/t5x/binary_search_test.py index 583a7e066..350a83787 100644 --- a/t5x/binary_search_test.py +++ b/t5x/binary_search_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/checkpoint_importer.py b/t5x/checkpoint_importer.py index 0233a99e2..8baa09f99 100644 --- a/t5x/checkpoint_importer.py +++ b/t5x/checkpoint_importer.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/checkpoint_importer_test.py b/t5x/checkpoint_importer_test.py index c1191f7e7..5ae0b8ad3 100644 --- a/t5x/checkpoint_importer_test.py +++ b/t5x/checkpoint_importer_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/checkpoint_utils.py b/t5x/checkpoint_utils.py index 60e517dbf..d0e428d60 100644 --- a/t5x/checkpoint_utils.py +++ b/t5x/checkpoint_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/checkpoint_utils_test.py b/t5x/checkpoint_utils_test.py index ba203f859..ca4fa5155 100644 --- a/t5x/checkpoint_utils_test.py +++ b/t5x/checkpoint_utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/checkpoints.py b/t5x/checkpoints.py index 286cc2465..724a217c9 100644 --- a/t5x/checkpoints.py +++ b/t5x/checkpoints.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/config_utils.py b/t5x/config_utils.py index 470d40f44..94c2b4f8c 100644 --- a/t5x/config_utils.py +++ b/t5x/config_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/configs/__init__.py b/t5x/configs/__init__.py index cb65e28b9..a52d4f952 100644 --- a/t5x/configs/__init__.py +++ b/t5x/configs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/configs/runs/__init__.py b/t5x/configs/runs/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/configs/runs/__init__.py +++ b/t5x/configs/runs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/__init__.py b/t5x/contrib/__init__.py index 2223fcaed..089ca7ab1 100644 --- a/t5x/contrib/__init__.py +++ b/t5x/contrib/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/calm/__init__.py b/t5x/contrib/calm/__init__.py index 2a2835d96..3283cf1d0 100644 --- a/t5x/contrib/calm/__init__.py +++ b/t5x/contrib/calm/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/calm/decoding.py b/t5x/contrib/calm/decoding.py index fe08ed462..93376c1bc 100644 --- a/t5x/contrib/calm/decoding.py +++ b/t5x/contrib/calm/decoding.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/calm/models.py b/t5x/contrib/calm/models.py index 3637cd840..a107dde3c 100644 --- a/t5x/contrib/calm/models.py +++ b/t5x/contrib/calm/models.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/__init__.py b/t5x/contrib/moe/__init__.py index e3726cadb..184349df1 100644 --- a/t5x/contrib/moe/__init__.py +++ b/t5x/contrib/moe/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/adafactor_utils.py b/t5x/contrib/moe/adafactor_utils.py index 70da27fd5..7af4eed72 100644 --- a/t5x/contrib/moe/adafactor_utils.py +++ b/t5x/contrib/moe/adafactor_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/checkpoints.py b/t5x/contrib/moe/checkpoints.py index 48ab94ca4..28244d9cc 100644 --- a/t5x/contrib/moe/checkpoints.py +++ b/t5x/contrib/moe/checkpoints.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/checkpoints_test.py b/t5x/contrib/moe/checkpoints_test.py index d1e22798c..672018c82 100644 --- a/t5x/contrib/moe/checkpoints_test.py +++ b/t5x/contrib/moe/checkpoints_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/configs/__init__.py b/t5x/contrib/moe/configs/__init__.py index cb65e28b9..a52d4f952 100644 --- a/t5x/contrib/moe/configs/__init__.py +++ b/t5x/contrib/moe/configs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/configs/runs/__init__.py b/t5x/contrib/moe/configs/runs/__init__.py index cb65e28b9..a52d4f952 100644 --- a/t5x/contrib/moe/configs/runs/__init__.py +++ b/t5x/contrib/moe/configs/runs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/models.py b/t5x/contrib/moe/models.py index 540ec8372..005b54b4d 100644 --- a/t5x/contrib/moe/models.py +++ b/t5x/contrib/moe/models.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/models_test.py b/t5x/contrib/moe/models_test.py index fa64fcd71..b16f25b70 100644 --- a/t5x/contrib/moe/models_test.py +++ b/t5x/contrib/moe/models_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/partitioning.py b/t5x/contrib/moe/partitioning.py index 99a183b6a..c4cd61d4c 100644 --- a/t5x/contrib/moe/partitioning.py +++ b/t5x/contrib/moe/partitioning.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/partitioning_test.py b/t5x/contrib/moe/partitioning_test.py index 31efe8aea..05c7b73be 100644 --- a/t5x/contrib/moe/partitioning_test.py +++ b/t5x/contrib/moe/partitioning_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/trainer.py b/t5x/contrib/moe/trainer.py index 566279e70..1a1a02241 100644 --- a/t5x/contrib/moe/trainer.py +++ b/t5x/contrib/moe/trainer.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/trainer_test.py b/t5x/contrib/moe/trainer_test.py index 3490ecfc2..704e9c2c5 100644 --- a/t5x/contrib/moe/trainer_test.py +++ b/t5x/contrib/moe/trainer_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/training_utils.py b/t5x/contrib/moe/training_utils.py index 7fbccce6c..98757e79f 100644 --- a/t5x/contrib/moe/training_utils.py +++ b/t5x/contrib/moe/training_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/contrib/moe/training_utils_test.py b/t5x/contrib/moe/training_utils_test.py index 0b47cd16b..bfae93f13 100644 --- a/t5x/contrib/moe/training_utils_test.py +++ b/t5x/contrib/moe/training_utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/decoding.py b/t5x/decoding.py index 5d4a8cff7..b71a3393d 100644 --- a/t5x/decoding.py +++ b/t5x/decoding.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/decoding_test.py b/t5x/decoding_test.py index d38e43f84..8ddef307f 100644 --- a/t5x/decoding_test.py +++ b/t5x/decoding_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/disable_gc_during_import.py b/t5x/disable_gc_during_import.py index 4b3749e54..470c6f103 100644 --- a/t5x/disable_gc_during_import.py +++ b/t5x/disable_gc_during_import.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/disable_gc_during_import_test.py b/t5x/disable_gc_during_import_test.py index d3fbe7359..48ebacd39 100644 --- a/t5x/disable_gc_during_import_test.py +++ b/t5x/disable_gc_during_import_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/eval.py b/t5x/eval.py index 8314fed90..f9a9c5d1f 100644 --- a/t5x/eval.py +++ b/t5x/eval.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/__init__.py b/t5x/examples/__init__.py index e4c38ff26..548e50465 100644 --- a/t5x/examples/__init__.py +++ b/t5x/examples/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/decoder_only/layers.py b/t5x/examples/decoder_only/layers.py index 4c21dec74..9980fec0c 100644 --- a/t5x/examples/decoder_only/layers.py +++ b/t5x/examples/decoder_only/layers.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/decoder_only/layers_test.py b/t5x/examples/decoder_only/layers_test.py index 971a11cf8..a68e35fe6 100644 --- a/t5x/examples/decoder_only/layers_test.py +++ b/t5x/examples/decoder_only/layers_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/decoder_only/network.py b/t5x/examples/decoder_only/network.py index 09dbda85e..bcddd95d6 100644 --- a/t5x/examples/decoder_only/network.py +++ b/t5x/examples/decoder_only/network.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/decoder_only/network_test.py b/t5x/examples/decoder_only/network_test.py index 23ccf0eab..15b4fd29f 100644 --- a/t5x/examples/decoder_only/network_test.py +++ b/t5x/examples/decoder_only/network_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/__init__.py b/t5x/examples/scalable_t5/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/__init__.py +++ b/t5x/examples/scalable_t5/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/layers.py b/t5x/examples/scalable_t5/layers.py index 7ba8e4545..969f158b0 100644 --- a/t5x/examples/scalable_t5/layers.py +++ b/t5x/examples/scalable_t5/layers.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/layers_test.py b/t5x/examples/scalable_t5/layers_test.py index b0d2251a0..41b736792 100644 --- a/t5x/examples/scalable_t5/layers_test.py +++ b/t5x/examples/scalable_t5/layers_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/mt5/__init__.py b/t5x/examples/scalable_t5/mt5/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/mt5/__init__.py +++ b/t5x/examples/scalable_t5/mt5/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/network.py b/t5x/examples/scalable_t5/network.py index 74e029e97..b20a7c56d 100644 --- a/t5x/examples/scalable_t5/network.py +++ b/t5x/examples/scalable_t5/network.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/network_test.py b/t5x/examples/scalable_t5/network_test.py index fab3c43fe..59daeeb2d 100644 --- a/t5x/examples/scalable_t5/network_test.py +++ b/t5x/examples/scalable_t5/network_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/t5_1_1/__init__.py b/t5x/examples/scalable_t5/t5_1_1/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/t5_1_1/__init__.py +++ b/t5x/examples/scalable_t5/t5_1_1/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/t5_1_1/examples/__init__.py b/t5x/examples/scalable_t5/t5_1_1/examples/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/t5_1_1/examples/__init__.py +++ b/t5x/examples/scalable_t5/t5_1_1/examples/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/umt5/__init__.py b/t5x/examples/scalable_t5/umt5/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/umt5/__init__.py +++ b/t5x/examples/scalable_t5/umt5/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/umt5/architectures/__init__.py b/t5x/examples/scalable_t5/umt5/architectures/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/umt5/architectures/__init__.py +++ b/t5x/examples/scalable_t5/umt5/architectures/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/umt5/optimizer/__init__.py b/t5x/examples/scalable_t5/umt5/optimizer/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/umt5/optimizer/__init__.py +++ b/t5x/examples/scalable_t5/umt5/optimizer/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/scalable_t5/umt5/runs/__init__.py b/t5x/examples/scalable_t5/umt5/runs/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/scalable_t5/umt5/runs/__init__.py +++ b/t5x/examples/scalable_t5/umt5/runs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/__init__.py b/t5x/examples/t5/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/t5/__init__.py +++ b/t5x/examples/t5/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/byt5/__init__.py b/t5x/examples/t5/byt5/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/t5/byt5/__init__.py +++ b/t5x/examples/t5/byt5/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/layers.py b/t5x/examples/t5/layers.py index 52e04f40f..cc2e6ac8b 100644 --- a/t5x/examples/t5/layers.py +++ b/t5x/examples/t5/layers.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/layers_test.py b/t5x/examples/t5/layers_test.py index f68c2f9ae..db802b316 100644 --- a/t5x/examples/t5/layers_test.py +++ b/t5x/examples/t5/layers_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/mt5/__init__.py b/t5x/examples/t5/mt5/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/t5/mt5/__init__.py +++ b/t5x/examples/t5/mt5/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/network.py b/t5x/examples/t5/network.py index 2ac67df00..475c71c20 100644 --- a/t5x/examples/t5/network.py +++ b/t5x/examples/t5/network.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/network_test.py b/t5x/examples/t5/network_test.py index 0d709e407..0e83d315d 100644 --- a/t5x/examples/t5/network_test.py +++ b/t5x/examples/t5/network_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/t5_1_0/__init__.py b/t5x/examples/t5/t5_1_0/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/t5/t5_1_0/__init__.py +++ b/t5x/examples/t5/t5_1_0/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/t5_1_1/__init__.py b/t5x/examples/t5/t5_1_1/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/t5/t5_1_1/__init__.py +++ b/t5x/examples/t5/t5_1_1/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/examples/t5/t5_1_1/examples/__init__.py b/t5x/examples/t5/t5_1_1/examples/__init__.py index c4f8b134e..0146a9b54 100644 --- a/t5x/examples/t5/t5_1_1/examples/__init__.py +++ b/t5x/examples/t5/t5_1_1/examples/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/export.py b/t5x/export.py index aede62362..2c832f7a0 100644 --- a/t5x/export.py +++ b/t5x/export.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/export_lib.py b/t5x/export_lib.py index dfa0fdccb..98e5d4ca6 100644 --- a/t5x/export_lib.py +++ b/t5x/export_lib.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/__init__.py b/t5x/fiddle_configs/__init__.py index 1234b1d61..eba3db493 100644 --- a/t5x/fiddle_configs/__init__.py +++ b/t5x/fiddle_configs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/configs/__init__.py b/t5x/fiddle_configs/configs/__init__.py index 1234b1d61..eba3db493 100644 --- a/t5x/fiddle_configs/configs/__init__.py +++ b/t5x/fiddle_configs/configs/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/configs/finetune.py b/t5x/fiddle_configs/configs/finetune.py index 602f020c9..8b78188f5 100644 --- a/t5x/fiddle_configs/configs/finetune.py +++ b/t5x/fiddle_configs/configs/finetune.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/configs/pretrain.py b/t5x/fiddle_configs/configs/pretrain.py index 943e05be4..f05b31c56 100644 --- a/t5x/fiddle_configs/configs/pretrain.py +++ b/t5x/fiddle_configs/configs/pretrain.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/examples/__init__.py b/t5x/fiddle_configs/examples/__init__.py index 1234b1d61..eba3db493 100644 --- a/t5x/fiddle_configs/examples/__init__.py +++ b/t5x/fiddle_configs/examples/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/examples/t5_1_1.py b/t5x/fiddle_configs/examples/t5_1_1.py index e01c87b75..7fd2fbfcb 100644 --- a/t5x/fiddle_configs/examples/t5_1_1.py +++ b/t5x/fiddle_configs/examples/t5_1_1.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/examples/t5_1_1_test.py b/t5x/fiddle_configs/examples/t5_1_1_test.py index 77ab0dc34..7e3428650 100644 --- a/t5x/fiddle_configs/examples/t5_1_1_test.py +++ b/t5x/fiddle_configs/examples/t5_1_1_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/models/__init__.py b/t5x/fiddle_configs/models/__init__.py index 1234b1d61..eba3db493 100644 --- a/t5x/fiddle_configs/models/__init__.py +++ b/t5x/fiddle_configs/models/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/fiddle_configs/models/t5_1_1.py b/t5x/fiddle_configs/models/t5_1_1.py index d43d41570..e1fa447d8 100644 --- a/t5x/fiddle_configs/models/t5_1_1.py +++ b/t5x/fiddle_configs/models/t5_1_1.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/gin_utils.py b/t5x/gin_utils.py index d1e47a6b3..ff2050599 100644 --- a/t5x/gin_utils.py +++ b/t5x/gin_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/gin_utils_test.py b/t5x/gin_utils_test.py index 06f49146c..1805787e7 100644 --- a/t5x/gin_utils_test.py +++ b/t5x/gin_utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/infer.py b/t5x/infer.py index 2e088a6db..89a4bfe87 100644 --- a/t5x/infer.py +++ b/t5x/infer.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/interactive_model.py b/t5x/interactive_model.py index 2762616ba..ea453d759 100644 --- a/t5x/interactive_model.py +++ b/t5x/interactive_model.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/losses.py b/t5x/losses.py index 6c68f5d0d..a5e365476 100644 --- a/t5x/losses.py +++ b/t5x/losses.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/losses_test.py b/t5x/losses_test.py index 5a1b7c908..0ad0823ab 100644 --- a/t5x/losses_test.py +++ b/t5x/losses_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/main.py b/t5x/main.py index b36423764..17bad3bb1 100644 --- a/t5x/main.py +++ b/t5x/main.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/metrics.py b/t5x/metrics.py index f2ecd1e3f..9f21fa031 100644 --- a/t5x/metrics.py +++ b/t5x/metrics.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/metrics_test.py b/t5x/metrics_test.py index 22ca6343d..34879a4ea 100644 --- a/t5x/metrics_test.py +++ b/t5x/metrics_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/models.py b/t5x/models.py index b58f5791e..7433ea2e6 100644 --- a/t5x/models.py +++ b/t5x/models.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/models_test.py b/t5x/models_test.py index 9a1d5cb53..fb65fc0d8 100644 --- a/t5x/models_test.py +++ b/t5x/models_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/optimizers.py b/t5x/optimizers.py index d1dbc399d..01ffd4552 100644 --- a/t5x/optimizers.py +++ b/t5x/optimizers.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/optimizers_test.py b/t5x/optimizers_test.py index f5dd16146..f9d130eb4 100644 --- a/t5x/optimizers_test.py +++ b/t5x/optimizers_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/partitioning.py b/t5x/partitioning.py index 7165ac7f1..0b57e67fd 100644 --- a/t5x/partitioning.py +++ b/t5x/partitioning.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/partitioning_test.py b/t5x/partitioning_test.py index 3a8b0c3a0..66e02febf 100644 --- a/t5x/partitioning_test.py +++ b/t5x/partitioning_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/precompile.py b/t5x/precompile.py index 62d9eb020..c01d5b2ee 100644 --- a/t5x/precompile.py +++ b/t5x/precompile.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/scripts/__init__.py b/t5x/scripts/__init__.py index e4c38ff26..548e50465 100644 --- a/t5x/scripts/__init__.py +++ b/t5x/scripts/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/scripts/convert_tf_checkpoint.py b/t5x/scripts/convert_tf_checkpoint.py index 8eecdaf64..e5a0caace 100644 --- a/t5x/scripts/convert_tf_checkpoint.py +++ b/t5x/scripts/convert_tf_checkpoint.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/scripts/xm_launch.py b/t5x/scripts/xm_launch.py index a13da3383..bd59ff5ec 100644 --- a/t5x/scripts/xm_launch.py +++ b/t5x/scripts/xm_launch.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/state_utils.py b/t5x/state_utils.py index bfde5f41f..aa1af4637 100644 --- a/t5x/state_utils.py +++ b/t5x/state_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/state_utils_test.py b/t5x/state_utils_test.py index 14eb2a9ec..3d2b1d8fb 100644 --- a/t5x/state_utils_test.py +++ b/t5x/state_utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/test_utils.py b/t5x/test_utils.py index badd76488..51a3b8bdd 100644 --- a/t5x/test_utils.py +++ b/t5x/test_utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/train.py b/t5x/train.py index e85d37c2a..727cda359 100644 --- a/t5x/train.py +++ b/t5x/train.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/train_state.py b/t5x/train_state.py index abcbc790e..04a86288e 100644 --- a/t5x/train_state.py +++ b/t5x/train_state.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/train_state_test.py b/t5x/train_state_test.py index cddea16dd..768a0112f 100644 --- a/t5x/train_state_test.py +++ b/t5x/train_state_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/trainer.py b/t5x/trainer.py index 3d592d8c3..965bd09a1 100644 --- a/t5x/trainer.py +++ b/t5x/trainer.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/trainer_test.py b/t5x/trainer_test.py index aca342a62..ecd708f21 100644 --- a/t5x/trainer_test.py +++ b/t5x/trainer_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/utils.py b/t5x/utils.py index 701cee90b..304d98ba9 100644 --- a/t5x/utils.py +++ b/t5x/utils.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -32,6 +32,7 @@ from absl import flags from absl import logging import airio +from airio.grain import dataset_providers from airio.grain.common import feature_converters import clu.data import flax @@ -1847,7 +1848,7 @@ def get_vocabulary( if isinstance(cfg.mixture_or_task_name, airio.DatasetProviderBase): mixture_or_task = cfg.mixture_or_task_name - vocab_map = airio.get_vocabularies(mixture_or_task) + vocab_map = dataset_providers.get_vocabularies(mixture_or_task) if not vocab_map: raise ValueError( f'No vocabularies found for AirIO task/mixture {mixture_or_task}' diff --git a/t5x/utils_test.py b/t5x/utils_test.py index bb4c07ea2..1d200c569 100644 --- a/t5x/utils_test.py +++ b/t5x/utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/t5x/version.py b/t5x/version.py index 4bd8cfcd5..9b7793020 100644 --- a/t5x/version.py +++ b/t5x/version.py @@ -1,4 +1,4 @@ -# Copyright 2023 The T5X Authors. +# Copyright 2024 The T5X Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License.