class DecoderProcessor extends AnyRef
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- DecoderProcessor
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Instance Constructors
- new DecoderProcessor(batchSize: Int, maxTextLength: Int, sequenceLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, vocabSize: Int, noRepeatNgramSize: Int, repetitionPenalty: Double, randomSeed: Option[Long], stopTokens: Array[Int], ignoreTokenIds: Array[Int], maxNewTokens: Int)
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
- val batchSize: Int
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
- def createNextTokenLogitsPenalties(inputIds: Seq[Array[Int]], logits: Array[Array[Float]], repetitionPenalty: Double): Array[Array[Float]]
- var currentLength: Int
- val doSample: Boolean
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
- val ignoreTokenIds: Array[Int]
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- val maxNewTokens: Int
- val maxTextLength: Int
- var nPredictedTokens: Int
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- val noRepeatNgramSize: Int
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
- def processLogits(batchLogits: Array[Array[Float]], decoderInputIds: Array[Array[Int]]): Array[Array[Int]]
- def processLogits(batchLogits: Array[Array[Float]], decoderInputIds: Array[Array[Long]]): Array[Array[Long]]
- val randomSeed: Option[Long]
- val repetitionPenalty: Double
- var sentenceLengths: List[Int]
- val sequenceLength: Int
- def stopDecoding(decoderInputIds: Array[Array[Long]]): Boolean
- def stopDecoding(decoderInputIds: Array[Array[Int]]): Boolean
- val stopTokens: Array[Int]
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
- val temperature: Double
-
def
toString(): String
- Definition Classes
- AnyRef → Any
- val topK: Int
- val topP: Double
- var unfinishedSentences: List[Int]
- val vocabSize: Int
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()