Package chat.octet.model.components.processor.impl

Class NoBadWordsLogitsProcessor

java.lang.Object

chat.octet.model.components.processor.impl.NoBadWordsLogitsProcessor

All Implemented Interfaces:

LogitsProcessor

public class NoBadWordsLogitsProcessor
extends Object
implements LogitsProcessor

Constructor Summary

Constructors

Constructor	Description
NoBadWordsLogitsProcessor(int[] badWordsTokenIds)

Method Summary

Modifier and Type	Method	Description
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

NoBadWordsLogitsProcessor

public NoBadWordsLogitsProcessor(int[] badWordsTokenIds)

Method Details

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.