4 ms·
If we check the script, it seems to support open ai models and llama https://github.com/lovishchopra/NL2FOL/blob/main/src/nl_to_fol.py https://github.com/lovish
by janalsncm 2y ago
If we check the script, it seems to support open ai models and llama https://github.com/lovishchopra/NL2FOL/blob/main/src/nl_to_fol.py https://github.com/lovishchopra/NL2FOL/blob/main/src/nl_to_f...
It would have been a lot cooler if this was set up as a pretrained model using RL to translate.
- rahimnathwani 2y agoThat's for --model_name, not --nli_model_name: https://github.com/lovishchopra/NL2FOL/blob/4635a81f216da2ad69c1d36215d6bd88841cc6d8/src/nl_to_fol.py#L459 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)