from transformers import AutoTokenizer, AutoModelForTokenClassification

# Load the model and tokenizer
tokenizer = AutoTokenizer.from_pretrained("your-username/your-model-name", trust_remote_code=True)
model = AutoModelForTokenClassification.from_pretrained("your-username/your-model-name", trust_remote_code=True)

# Use for inference
inputs = tokenizer(["Your text here"], return_tensors="pt", is_split_into_words=False)
outputs = model(**inputs)
predictions = outputs.logits.argmax(dim=-1)

Training Details

Training Data: ['automated-analytics/ai4privacy-pii-masking-en-v1-ner-coarse', 'automated-analytics/gretel-pii-masking-en-v1-ner-coarse']
Learning Rate: 5e-05
Batch Size: 128
Epochs: 3
Max Length: 128

Important Note

This model uses a custom model class. Make sure to use trust_remote_code=True when loading the model.

automated-analytics/piiceetah-unmask-qwen2.5-0.5b

UnmaskingQwen3 for Token Classification

Model Details

Usage

Training Details

Important Note

automated-analytics/
piiceetah-unmask-qwen2.5-0.5b