# clean up
del pipe
del model
del tokenizer
torch.cuda.empty_cache()