TextGenerator

java.lang.Object

io.github.kirstenali.deepj.models.gpt.TextGenerator

public final class TextGenerator extends Object

Simple autoregressive text generation for GPTModel. Byte-level tokens, temperature sampling, and optional top-k sampling.

Method Summary

Modifier and Type

Method

Description

static String

generate(GPTModel model, Tokenizer tok, GPTConfig cfg, String prompt, int maxNewTokens, double temperature, int topK, long seed)

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details
- generate
  
  public static String generate(GPTModel model, Tokenizer tok, GPTConfig cfg, String prompt, int maxNewTokens, double temperature, int topK, long seed)