Package chat.octet.model.components.processor

Class LogitsProcessorList

java.lang.Object

chat.octet.model.components.processor.LogitsProcessorList

All Implemented Interfaces:

LogitsProcessor

public final class LogitsProcessorList
extends Object
implements LogitsProcessor

Stopping criteria list

Author:

William

Constructor Summary

Constructors

Constructor	Description
LogitsProcessorList()

Method Summary

Modifier and Type	Method	Description
LogitsProcessorList	add(LogitsProcessor processor)
boolean	isEmpty()
float[]	processor(int[] inputTokenIds, float[] scores, Object... args)	Logits processor

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

LogitsProcessorList

public LogitsProcessorList()

Method Details

add

public LogitsProcessorList add(LogitsProcessor processor)

isEmpty

public boolean isEmpty()

processor

public float[] processor(@Nullable
 int[] inputTokenIds,
 @Nonnull
 float[] scores,
 Object... args)

Description copied from interface: LogitsProcessor

Logits processor

Specified by:

processor in interface LogitsProcessor

Parameters:

inputTokenIds - Indices of input sequence tokens in the vocabulary.

scores - Prediction scores of a language modeling head. These can be logits for each vocabulary.

args - Specific args to a logits processor.

Returns:

float[] The processed prediction scores.