Value Members

1.  final def !=(arg0: Any): Boolean

   Definition Classes

   AnyRef → Any

2.  final def ##(): Int

   Definition Classes

   AnyRef → Any

3.  final def ==(arg0: Any): Boolean

   Definition Classes

   AnyRef → Any

4.  val additionalTokens: Map[Int, String]

   Definition Classes

   OnnxT5EncoderDecoder → T5EncoderDecoder

5.  final def asInstanceOf[T0]: T0

   Definition Classes

   Any

6.  def clone(): AnyRef

   Attributes

   protected[lang]

   Definition Classes

   AnyRef

   Annotations

   @throws( ... ) @native()

7.  def decode(sentences: Array[Array[Int]]): Seq[String]

   Attributes

   protected

   Definition Classes

   T5EncoderDecoder

8.  lazy val decoderCacheInputKeys: Array[String]
9.  lazy val decoderCacheOutputKeys: Array[String]
10.  def encode(sentences: Seq[Annotation], task: String): Seq[Array[Int]]

    Attributes

    protected

    Definition Classes

    T5EncoderDecoder

11.  lazy val encoderCacheInputKeys: Array[String]
12.  lazy val encoderCacheOutputKeys: Array[String]
13.  val eosTokenId: Int

    Attributes

    protected

    Definition Classes

    T5EncoderDecoder

14.  final def eq(arg0: AnyRef): Boolean

    Definition Classes

    AnyRef

15.  def equals(arg0: Any): Boolean

    Definition Classes

    AnyRef → Any

16.  def finalize(): Unit

    Attributes

    protected[lang]

    Definition Classes

    AnyRef

    Annotations

    @throws( classOf[java.lang.Throwable] )

17.  def generate(prompts: Seq[Annotation], batchSize: Int, maxNewTokens: Int, maxContextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, randomSeed: Option[Long], ignoreTokenIds: Array[Int], isCaseSensitive: Boolean, stopAtEos: Boolean, noRepeatNgramSize: Int, repetitionPenalty: Double): Seq[Annotation]

    Definition Classes

    T5EncoderDecoder

18.  def generateCacheKeys(component: String, state: String): Array[String]
19.  def generateNoBeamSearch(inputIds: Seq[Array[Int]], encoderStateTensors: OnnxTensor, encoderAttentionMaskTensors: OnnxTensor, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, vocabSize: Int, randomSeed: Option[Long], ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean, noRepeatNgramSize: Int, repetitionPenalty: Double): Array[Array[Int]]
20.  final def getClass(): Class[_]

    Definition Classes

    AnyRef → Any

    Annotations

    @native()

21.  def hashCode(): Int

    Definition Classes

    AnyRef → Any

    Annotations

    @native()

22.  final def isInstanceOf[T0]: Boolean

    Definition Classes

    Any

23.  final def ne(arg0: AnyRef): Boolean

    Definition Classes

    AnyRef

24.  final def notify(): Unit

    Definition Classes

    AnyRef

    Annotations

    @native()

25.  final def notifyAll(): Unit

    Definition Classes

    AnyRef

    Annotations

    @native()

26.  val numAttnHeads: Int

    Attributes

    protected

27.  val numLayers: Int

    Attributes

    protected

28.  val onnxDecoder: OnnxWrapper
29.  val onnxEncoder: OnnxWrapper
30.  val paddingTokenId: Int

    Attributes

    protected

    Definition Classes

    T5EncoderDecoder

31.  val pieceSize: Int

    Attributes

    protected

    Definition Classes

    T5EncoderDecoder

32.  def predict(sentences: Seq[Annotation], task: String, batchSize: Int, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, randomSeed: Option[Long] = None, ignoreTokenIds: Array[Int] = Array(), isCaseSensitive: Boolean, stopAtEos: Boolean, noRepeatNgramSize: Int, repetitionPenalty: Double): Seq[Annotation]

    Definition Classes

    T5EncoderDecoder

33.  def sessionWarmup(): Unit

    Definition Classes

    T5EncoderDecoder

34.  val spp: SentencePieceWrapper

    Definition Classes

    OnnxT5EncoderDecoder → T5EncoderDecoder

35.  final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes

    AnyRef

36.  def tag(batch: Seq[Array[Int]], maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, noRepeatNgramSize: Int, repetitionPenalty: Double, randomSeed: Option[Long], ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean): Array[Array[Int]]

    Definition Classes

    OnnxT5EncoderDecoder → T5EncoderDecoder

37.  def toString(): String

    Definition Classes

    AnyRef → Any

38.  val vocabSize: Int

    Attributes

    protected

    Definition Classes

    T5EncoderDecoder

39.  final def wait(): Unit

    Definition Classes

    AnyRef

    Annotations

    @throws( ... )

40.  final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes

    AnyRef

    Annotations

    @throws( ... )

41.  final def wait(arg0: Long): Unit

    Definition Classes

    AnyRef

    Annotations

    @throws( ... ) @native()