Skip to content

Commit 4f15b03

Browse files
SiqiaoWu1993tensorflow-copybara
authored andcommitted
Internel change only
PiperOrigin-RevId: 831644100
1 parent 1c88766 commit 4f15b03

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tensorflow_serving/servables/tensorflow/tfrt_saved_model_source_adapter.proto

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,10 @@ message TfrtSavedModelConfig {
244244
bool allow_orbax = 2030;
245245
// copybara:strip_end
246246
bool allow_xla_cpu = 2031;
247+
248+
// The number of background threads to use for loading checkpoints, or 0 to
249+
// disable asynchronous checkpoint loading.
250+
int64 async_checkpoint_loading_num_threads = 2032;
247251
}
248252

249253
// Config proto for TfrtSavedModelSourceAdapter.

0 commit comments

Comments
 (0)