Skip to content

Commit 15cafce

Browse files
No public description
PiperOrigin-RevId: 639121310
1 parent 778e52f commit 15cafce

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

official/common/flags.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,3 +112,12 @@ def define_flags():
112112

113113
flags.DEFINE_string(
114114
'tpu_platform', default=None, help='TPU platform type.')
115+
116+
flags.DEFINE_string(
117+
'tfhub_handle',
118+
None,
119+
'TFHub handle for publishing the model to TFHub. The model '
120+
'is published to TFHub if this flag is set. Please use a '
121+
'TFHubPusher (go/tflex/standard_components/pusher) component if '
122+
'running in TFleX.',
123+
)

0 commit comments

Comments
 (0)