diff --git a/pyproject.toml b/pyproject.toml index 2841cae0b..f07ff3864 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,7 +29,7 @@ dependencies = [ "equinox>=0.11.7", "jaxtyping>=0.2.34", "tokenizers>=0.15.2", - "transformers>=4.41.2,<4.48.0", + "transformers>=4.41.2,<4.49.0", "optax>=0.1.9", "wandb>=0.17.8", "draccus>=0.9.3", diff --git a/src/levanter/__init__.py b/src/levanter/__init__.py index f9570aaf7..6b2cbeb1e 100644 --- a/src/levanter/__init__.py +++ b/src/levanter/__init__.py @@ -3,6 +3,7 @@ import levanter.data as data import levanter.distributed as distributed import levanter.eval as eval +import levanter.eval_harness as eval_harness import levanter.models as models import levanter.optim as optim import levanter.tracker as tracker