Number of virtual tokens to be used for training. In prompt tuning we are essentially learning the embedded representations for soft prompts, which are known as virtual tokens, via back propagation for a specific task(s) while keeping the rest of the model fixed. num_virtual_tokens is the number of dimensions for these virtual tokens.

interface TrainingNumVirtualTokens {
    default?: number;
    supported?: number[];
}

Properties

Properties

default?: number

The default number of virtual tokens.

supported?: number[]

The possible values for the number of virtual tokens.