cerebras.modelzoo.data.nlp.gpt.InferenceDataProcessor

Functions

get_token_ids	Get encoded token ids from a string using the specified tokenizer.
tokenize_stop_words	Helper to construct a list of stop token sequences from the given list of stop words using the specified tokenizer.

Classes

EvalHarnessDataset
InferenceDataProcessor
InferenceDataProcessorBCEH	Subclass for processing BigCode data, i.e. bigcode_eh requests.
InferenceDataProcessorGU	Subclass for processing EEH generate_until requests.
InferenceDataProcessorLL	Subclass for processing EEH loglikelihood requests.
RequestType	An enumeration.

previous

cerebras.modelzoo.data.nlp.gpt.HuggingFaceIterableDataProcessorEli5.HuggingFaceIterableDataProcessorEli5

next

cerebras.modelzoo.data.nlp.gpt.InferenceDataProcessor.get_token_ids