It would have been a lot cooler if this was set up as a pretrained model using RL to translate.
https://github.com/lovishchopra/NL2FOL/blob/4635a81f216da2ad...
nli_tokenizer = AutoTokenizer.from_pretrained(args.nli_model_name) nli_model = AutoModelForSequenceClassification.from_pretrained(args.nli_model_name)
It would have been a lot cooler if this was set up as a pretrained model using RL to translate.