• Home
• com.johnsnowlabs.nlp
• spark-nlp_2.12
• 5.5.1
• source code
• GenerationConfig.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.johnsnowlabs.ml.ai.util.Generation.GenerationConfig.scala Maven / Gradle / Ivy

package com.johnsnowlabs.ml.ai.util.Generation

case class GenerationConfig(
    bosId: Int,
    padId: Int,
    eosId: Int,
    vocabSize: Int,
    beginSuppressTokens: Option[Array[Int]],
    suppressTokenIds: Option[Array[Int]],
    forcedDecoderIds: Option[Array[(Int, Int)]])