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
Installation:
Installation Guide
API Reference:
ls_mlkit
ls_mlkit package
Subpackages
ls_mlkit.dataset package
ls_mlkit.diffuser package
ls_mlkit.model package
ls_mlkit.optimizer package
ls_mlkit.pipeline package
ls_mlkit.scheduler package
ls_mlkit.util package
Module contents
Related Topics
Documentation overview
ls_mlkit
ls_mlkit package
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