Fuck padding

models/gpt2.py

@@ -12,7 +12,7 @@ def load():
 
 def generate(input_text):
   # Tokenize the input text
-  input_ids = tokenizer.encode(input_text, return_tensors="pt", truncation=True, padding=True)
+  input_ids = tokenizer.encode(input_text, return_tensors="pt", truncation=True)
   attention_mask = tf.ones_like(input_ids)
   
   # Generate output using the model