Skip to content

Commit 717fd15

Browse files
Pathways-on-Cloud Teamcopybara-github
authored andcommitted
Clarify flag description for tpu_count.
PiperOrigin-RevId: 846790806
1 parent d4bd5c1 commit 717fd15

2 files changed

Lines changed: 1 addition & 2 deletions

File tree

pathwaysutils/experimental/shared_pathways_service/run_connect_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
flags.DEFINE_string(
2424
"tpu_type", "tpuv6e:2x2", "The TPU machine type and topology."
2525
)
26-
flags.DEFINE_integer("tpu_count", 1, "The number of TPU instances.")
26+
flags.DEFINE_integer("tpu_count", 1, "The number of TPU slices.")
2727

2828
flags.mark_flags_as_required([
2929
"cluster",

pathwaysutils/sidecar/python/requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,3 @@ jax>=0.5.1
33
tensorflow-datasets
44
tiktoken
55
grain-nightly>=0.0.1
6-
portpicker

0 commit comments

Comments
 (0)