While ChatGPT doesn't work in my geo.. I stumbled across the microsoft/codereviewer model. t = import fmt import go", truncation=True, padding=True, return tensors="pt") model.generate(t.input ids) ... ['I think this should be `fmt`']