Indus-Craft-VLM

nvidia/Llama-3.1-Nemotron-Nano-VL-8B-V1 fine-tuned (LoRA on the language model + trained vision projector, then merged) to caption images of traditional Indian crafts and textiles in 11 languages: English, Hindi, Marathi, Bengali, Urdu, Nepali, Tamil, Telugu, Dogri, Kashmiri, Maithili. Training data: vedantjadhav701/Indus-Craft-VLM (dataset repo, private).

Usage

Needs trust_remote_code=True. The model class only accepts eager at load time, so switch the LLM to SDPA afterwards:

import torch
from transformers import AutoModel, AutoConfig, AutoTokenizer, AutoImageProcessor
repo = "vedantjadhav701/Indus-Craft-VLM"
cfg = AutoConfig.from_pretrained(repo, trust_remote_code=True)
cfg.llm_config._attn_implementation = "sdpa"; cfg.vision_config.use_flash_attn = False
model = AutoModel.from_pretrained(repo, config=cfg, attn_implementation="eager", torch_dtype=torch.bfloat16,
                                  trust_remote_code=True).cuda().eval()
model.language_model.config._attn_implementation = "sdpa"
tok = AutoTokenizer.from_pretrained(repo); proc = AutoImageProcessor.from_pretrained(repo, trust_remote_code=True)

Prompts used in training, e.g.: Describe this traditional Indian craft image in detail in Tamil. (also: one-sentence caption, two-sentence summary, and "Identify the craft shown in this image..."). See 05_infer.py for batched generation.

Limitations

  • English captions were generated by Qwen3-VL-30B-A3B from the images and the other 10 languages are machine translations of them; none were human-verified. Dogri, Kashmiri and Maithili are the weakest. Have native speakers review before relying on them.
  • The base model's card lists English only; the Llama-3.1 tokenizer is inefficient on Indic scripts (long outputs, slow generation).
  • Captions describe visible materials/motifs; craft names and regions come from dataset labels and can be wrong for look-alike crafts.

Licence: inherits the NVIDIA Open Model License and the Llama 3.1 Community License. Built with Llama.

Downloads last month
5
Safetensors
Model size
9B params
Tensor type
I64
ยท
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for vedantjadhav701/Indus-Craft-VLM

Adapter
(2)
this model