def tokenize_prompt()

in training/train_amused.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def tokenize_prompt(tokenizer, prompt):
    return tokenizer(
        prompt,
        truncation=True,
        padding="max_length",
        max_length=77,
        return_tensors="pt",
    ).input_ids