This is model for LLM4FIN, including two models based on rule filtering model and rule element extraction model.
model = AutoModelForSequenceClassification.from_pretrained("rule_filtering", num_labels=3)
tokenizer = AutoTokenizer.from_pretrained("rule_filtering")
to load the model and tokenizer.
model = AutoModelForTokenClassification.from_pretrained("rule_element_extraction", num_labels=num_labels)
tokenizer = AutoTokenizer.from_pretrained("rule_element_extraction")
to load the model and tokenizer.
