tasks-gallery.json

[
    {
        "title": "Evaluation Task Example",
        "description": "A sample evaluation task that runs an evaluation script and creates reports.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "demo-eval-task",
        "metadata": {
            "category": "eval",
            "modality": "text",
            "framework": [
                "huggingface"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "AMD": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "AppleSilicon": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "cpu": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            }
        }
    },
    {
        "title": "AutoTrain SFT Training",
        "description": "SFT training task using Hugging Face AutoTrain. Requires a Hugging Face token in HF_TOKEN env var for model access.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "autotrain-sft",
        "metadata": {
            "category": "finetuning",
            "modality": "text",
            "framework": [
                "huggingface"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "Unsloth LLM Fine-tuning",
        "description": "LLM training task using Unsloth FastLanguageModel with LoRA fine-tuning. Requires a Hugging Face token in HF_TOKEN env var for model access.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "unsloth-llm-train",
        "metadata": {
            "category": "finetuning",
            "modality": "text",
            "framework": [
                "unsloth"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "Unsloth GRPO Training Task",
        "description": "A GRPO trainer based on the unsloth grpo training notebooks for training models with reasoning capabilities.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "unsloth-grpo-train",
        "metadata": {
            "category": "training",
            "modality": "text",
            "framework": [
                "unsloth"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "Unsloth Text-to-Speech Training Task",
        "description": "A Text-to-Speech (TTS) trainer based on the unsloth audio training notebooks.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "unsloth-text-to-speech-train",
        "metadata": {
            "category": "training",
            "modality": "audio",
            "framework": [
                "unsloth"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "Diffusion LoRA Trainer",
        "description": "A LoRA trainer for Stable Diffusion models. Trains on images with captions and generates sample images before and after training.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "diffusion-train",
        "metadata": {
            "category": "training",
            "modality": "image",
            "framework": [
                "diffusers"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "Embedding Model Trainer",
        "description": "A trainer for embedding models using Sentence Transformers. Supports various dataset types and loss functions for training high-quality embeddings.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "embedding-model-train",
        "metadata": {
            "category": "training",
            "modality": "embedding",
            "framework": [
                "sentence-transformers"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "Sample Task Example",
        "description": "A sample PyTorch training task that clones a repository and runs training script",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "new-task-test",
        "metadata": {
            "category": "other",
            "modality": "other",
            "framework": []
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "AMD": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "AppleSilicon": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "cpu": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            }
        }
    },
    {
        "title": "Dataset Generation Task Example",
        "description": "A sample dataset generation task that runs a dataset generation script and creates a dataset.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "demo-generate-task",
        "metadata": {
            "category": "dataset-generation",
            "modality": "other",
            "framework": []
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "AMD": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "AppleSilicon": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            },
            "cpu": {
                "resources": {
                    "cpus": "2",
                    "memory": "4"
                }
            }
        }
    },
    {
        "title": "YOLO Object Detection Train Task Example",
        "description": "A sample YOLO object detection train task that runs a YOLO object detection train script on COCO8 dataset for 10 epochs",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "yolo-train-task",
        "metadata": {
            "category": "training",
            "modality": "vision",
            "framework": [
                "yolo"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
      "title": "Batch Image Generation using FLUX2",
      "description": "A task to generate images in batch using the FLUX2 model.",
      "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
      "github_repo_dir": "batch-image-gen-flux",
      "metadata": {
        "category": "generation",
        "modality": "vision",
        "framework": [""]
      },
      "supportedAccelerators": {
        "NVIDIA": {
          "resources": {
            "accelerators": "RTX3090:1"
          }
        },
        "AMD": {
          "resources": {
            "accelerators": "7900XTX:1"
          }
        }
      }
    },
    {
        "title": "MNIST Train Task",
        "description": "A task to train a neural network on the MNIST dataset using PyTorch.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples/",
        "github_repo_dir": "mnist-train",
        "metadata": {
            "category": "training",
            "modality": "other",
            "framework": []
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "MNIST JAX Train Task",
        "description": "A task to train a neural network on the MNIST dataset using JAX and Flax.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples/",
        "github_repo_dir": "mnist-jax-train",
        "metadata": {
            "category": "training",
            "modality": "other",
            "framework": [
                "jax",
                "flax"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    },
    {
        "title": "HF Training with Evals",
        "description": "A task to train a language model using the TRL library on a specified dataset with evaluation done using the EleutherAI Harness.",
        "github_repo_url": "https://github.com/transformerlab/transformerlab-examples",
        "github_repo_dir": "trl-training-evals",
        "metadata": {
            "category": "finetuning",
            "modality": "text",
            "framework": [
                "huggingface"
            ]
        },
        "supportedAccelerators": {
            "NVIDIA": {
                "resources": {
                    "accelerators": "RTX3090:1"
                }
            },
            "AMD": {
                "resources": {
                    "accelerators": "7900XTX:1"
                }
            }
        }
    }
]