How to use ai4data/datause-extraction-v2 with PEFT:
from peft import PeftModel from transformers import AutoModel base_model = AutoModel.from_pretrained("fastino/gliner2-large-v1") model = PeftModel.from_pretrained(base_model, "ai4data/datause-extraction-v2")
How to use ai4data/datause-extraction-v2 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("ai4data/datause-extraction-v2", device_map="auto")
How to use ai4data/datause-extraction-v2 with GLiNER2:
from gliner2 import AutoExtractor extractor = AutoExtractor.from_pretrained("ai4data/datause-extraction-v2") # Extract entities text = "Apple CEO Tim Cook announced iPhone 15 in Cupertino yesterday." result = extractor.extract_entities(text, ["company", "person", "product", "location"]) print(result)
