diff --git a/pathwaysutils/experimental/shared_pathways_service/run_connect_example.py b/pathwaysutils/experimental/shared_pathways_service/run_connect_example.py index 1835571..e2092cd 100644 --- a/pathwaysutils/experimental/shared_pathways_service/run_connect_example.py +++ b/pathwaysutils/experimental/shared_pathways_service/run_connect_example.py @@ -23,7 +23,7 @@ flags.DEFINE_string( "tpu_type", "tpuv6e:2x2", "The TPU machine type and topology." ) -flags.DEFINE_integer("tpu_count", 1, "The number of TPU instances.") +flags.DEFINE_integer("tpu_count", 1, "The number of TPU slices.") flags.mark_flags_as_required([ "cluster", diff --git a/pathwaysutils/sidecar/python/requirements.txt b/pathwaysutils/sidecar/python/requirements.txt index 517a98a..6606607 100644 --- a/pathwaysutils/sidecar/python/requirements.txt +++ b/pathwaysutils/sidecar/python/requirements.txt @@ -3,4 +3,3 @@ jax>=0.5.1 tensorflow-datasets tiktoken grain-nightly>=0.0.1 -portpicker