ls_mlkit.model.decoder_tf.generate module
¶
ls_mlkit.model.decoder_tf.generate.
greedy_decode
(
model
,
tokenizer
,
prompt
,
max_len
,
device
)
[source]
¶
ls-mlkit
Navigation
API Reference:
ls_mlkit.dataset package
ls_mlkit.diffusion package
ls_mlkit.model package
ls_mlkit.optimizer package
ls_mlkit.pipeline package
ls_mlkit.scheduler package
ls_mlkit.util package
Related Topics
Documentation overview
ls_mlkit.model package
ls_mlkit.model.decoder_tf package
Previous:
ls_mlkit.model.decoder_tf.causal_transformer module
Next:
ls_mlkit.model.decoder_tf.tokenizer module