This repository contains the fine-tuned student model used by the
Intent-to-Workflow component of HieraMind.
Given a natural-language service request and a retrieved service catalog
context, the model generates a structured workflow DAG containing the
required service functions and their execution dependencies.
HieraMind is a hierarchical framework for intent-aware service orchestration
across heterogeneous network and computing infrastructures.
The model was fine-tuned using group-based reinforcement learning with
teacher-guided workflow evaluation. Semantic retrieval is used to restrict
the service catalog provided to the model before workflow generation.
Further end-to-end and ablation results are reported in the HieraMind paper.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_id = "mamadzebal/HieraMind-I2W-Qwen3-4B"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype="auto",
9 device_map="auto",
10)