How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("image-text-to-text", model="aitf-komdigi/KomdigiITS-8B-DFK-TextClassification-Final")
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"},
            {"type": "text", "text": "What animal is on the candy?"}
        ]
    },
]
pipe(text=messages)
# Load model directly
from transformers import AutoProcessor, AutoModelForMultimodalLM

processor = AutoProcessor.from_pretrained("aitf-komdigi/KomdigiITS-8B-DFK-TextClassification-Final")
model = AutoModelForMultimodalLM.from_pretrained("aitf-komdigi/KomdigiITS-8B-DFK-TextClassification-Final", device_map="auto")
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"},
            {"type": "text", "text": "What animal is on the candy?"}
        ]
    },
]
inputs = processor.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

Dataset Details

Tasks & Capabilities

  • Text Classification DFK (Disinformasi, Fitnah, Ujaran Kebencian)

Dataset Splits

  • Train Samples: 20,790
  • Validation Samples: 2,599
  • Testing Samples: 2,599

Target Labels

  1. DISINFORMASI
  2. FITNAH
  3. UJARAN KEBENCIAN
  4. FAKTA
  5. BUKAN DFK

Training Configurations

Supervised Fine-Tuning (SFT) Pipeline Parameters

Parameter Value
Max Sequence Length 2048 tokens
Batch Size 2
Gradient Accumulation Steps 16
Effective Batch Size 32
Learning Rate (LR) 2e-4
LR Scheduler Linear
Optimizer AdamW 8-bit
Number of Epochs 1
Warmup Steps 20
Weight Decay 0.01
Max Gradient Norm 1.0
Evaluation Steps Every 200 steps

LoRA (PEFT) Hyperparameters

Parameter Value
r (Rank) 16
lora_alpha 32
lora_dropout 0
bias none
use_rslora False
Gradient Checkpointing unsloth
Target Modules q_proj, k_proj, v_proj, o_proj, gate_proj, down_proj, up_proj

Uploaded model

  • Developed by: hnuka
  • License: apache-2.0
  • Finetuned from model : aitf-komdigi/KomdigiITS-8B-DFK-CPT

This mistral3 model was trained 2x faster with Unsloth

Downloads last month
8
Safetensors
Model size
9B params
Tensor type
F16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for aitf-komdigi/KomdigiITS-8B-DFK-TextClassification-Final

Collection including aitf-komdigi/KomdigiITS-8B-DFK-TextClassification-Final