Inference
Generate Text from A Pretrained Model
Learn about the generate_from_pretrained
function within Simplifine’s Inference Tools
def generate_from_pretrained(
model:AutoModelForCausalLM,
tokenizer:AutoTokenizer,
generate_confg:GenerationConfig,
data:dict={}
):
model
AutoModelForCausalLM
requiredThe pre-trained causal language model to use for text generation.
tokenizer
AutoTokenizer
requiredThe tokenizer corresponding to the pre-trained model.
generate_confg
GenerationConfig
requiredThe configuration settings for text generation.
data
dict
A dictionary containing any additional data required for text generation. Defaults to an empty dictionary.