ai4science-studio: Skill for Cursor

Running models in AI4Science Studio

Use this skill when a user asks to run, execute, or launch any model in the repository.

Step 0: Check for cluster config

Before anything else, check if a cluster config exists:

# Check both locations
test -f .cluster-config.yaml && echo "repo-local" || \
test -f ~/.config/ai4science-studio/cluster.yaml && echo "user-level" || \
echo "missing"

If missing, tell the user: "No cluster configuration found. Running /init-cluster to detect your cluster environment." Then run the init-cluster flow (auto-discover GPU, SLURM, containers, paths) before proceeding.

If a config exists, read it and use its values as defaults for SLURM partition, account, container runtime, GPU arch, and scratch paths throughout the run. Still confirm with the user if any value is empty or looks stale.

Step 1: Identify the model

1. Read models.yaml at the repo root to find the model by name, slug, or HF id.
2. Read <domain>/models/<slug>/model.yaml for full metadata: recipes, env vars, container image, hardware requirements.

Step 2: Determine the task

Match the user's request to a recipe in model.yaml:

• "run inference" / "predict" / "generate" → look for task: inference
• "train" / "fine-tune" / "pair-tune" → look for task: train or task: finetune
• "ensemble" → look for task: ensemble

Step 3: Ask the user for required inputs

Read the env_vars section of model.yaml. For every variable marked required: true that has default: null, ask the user for the value. Present the question with the variable's description.

For common patterns, always offer three options — provide manually, generate/build, or auto-discover:

• SIF path → "Do you have an Apptainer SIF file? (Yes / No — I'll generate the pull command / Auto-discover — I'll search the filesystem)"
• Overlay → "Do you have a pre-built overlay? (Yes / No, build one / No, skip overlay / Auto-discover — I'll search for an existing one)"
• Upstream repo → "Do you have the repo cloned? (Yes / No — I'll generate the clone command / Auto-discover — I'll search the filesystem)"
• SLURM partition/account → "How should I determine your partition/account? (Provide manually / Auto-discover — I'll query SLURM)"