thehardwarestore.ai

where AI agents shop for the right tools

── 98829 repos indexed ── | filtered by tag: #stable-diffusion
mudler/LocalAI ★ 49287 4474 APPLICATION

LocalAI is a modular AI inference server providing OpenAI-compatible APIs for running diverse models (LLMs, vision, audio, video) locally. It features a polyglot architecture with Go core services, pluggable backends (C++/Go/Python), and a React-based web UI. The system supports distributed deployment via NATS/PostgreSQL, model gallery management, and multimodal processing. Key capabilities include model loading from Hugging Face, quantization, PII detection, and real-time transcription. The codebase demonstrates strong engineering practices with dependency injection, middleware chains, and clean separation of concerns.

RATING: 9.8/10 MAINT: active DRIFT: low
▸ 4 reusable components found
unslothai/unsloth ★ 77151 7091 APPLICATION

Unsloth is a full-stack, local-first AI platform for fine-tuning and inference of large language models (LLMs), diffusion models, and audio models. It consists of two primary components: (1) the core `unsloth` Python library, which provides highly optimized LoRA/QLoRA/RL training with custom Triton kernels, and (2) the Unsloth Studio application, a Tauri desktop app with a React frontend (TanStack Router, Zustand) and a FastAPI backend (SQLite, structlog). The platform enables users to discover models from the Hugging Face Hub, download and manage them locally, run inference with chat, image/video generation, and RAG, fine-tune with a visual recipe editor, and export in multiple formats (GGUF, NVFP4). It also integrates external LLM providers (OpenAI, Anthropic, Ollama) and supports agentic tool loops via MCP and Codex transports. The codebase is extensively tested with both unit tests and source-level contract tests, and includes cross-platform desktop installers for Windows, macOS, and Linux.

RATING: 9.6/10 MAINT: active DRIFT: low
▸ 5 reusable components found
huggingface/diffusers ★ 34684 7379 APPLICATION

The Hugging Face Diffusers library is a comprehensive, production-grade Python library for diffusion model inference and fine-tuning. It provides a vast collection of pre-built pipelines (text-to-image, image-to-image, inpainting, video generation, audio generation, and more), model architectures (UNet, Transformers, Autoencoders), schedulers (DDPM, DDIM, Flow Matching, etc.), and a rich ecosystem of loaders (LoRA, IP-Adapter, textual inversion, single-file), optimizers (attention backends, caching, offloading, tensor parallelism), and guiders (CFG, PAG, MAG, TCFG). The architecture is highly modular: a central `DiffusionPipeline` base class with mixins for loading and configuration, alongside a newer `ModularPipeline` framework that composes generation from reusable, declarative blocks (encoders, denoisers, decoders). The codebase is actively maintained with extensive test coverage (over 200k lines of tests), CI/CD pipelines, and support for quantization (bitsandbytes, GGUF, torchao). It is the de facto standard library for state-of-the-art diffusion models in PyTorch.

RATING: 9.6/10 MAINT: active DRIFT: low
▸ 4 reusable components found
invoke-ai/InvokeAI ★ 28537 3024 APPLICATION

InvokeAI is a production-grade, open-source creative engine for AI-assisted image and video generation. It provides a web-based user interface (with two frontend generations: a legacy webv1 and a current webv2) and a command-line interface. The backend is built with Python and FastAPI, using a node-based invocation system that allows users to compose generation pipelines from modular building blocks (e.g., denoising, VAE encoding/decoding, controlnet, IP-Adapter, LoRA patching). The system supports a wide range of model architectures including Stable Diffusion (1.5, 2.0, XL, 3.5), FLUX, Ideogram, CogView4, Z-Image, Krea, Anima, Qwen, ERNIE, Wan, MiniMax H3, and LTX-2, as well as external API providers like OpenAI and Gemini. The application includes a sophisticated canvas editor with layers (raster, control, inpaint mask, regional guidance), compositing operations, and undo/redo; a visual workflow editor for constructing node graphs; a gallery with boards, search, and metadata recall; a model manager for installing, configuring, and converting models; a queue system for batch processing with progress tracking and cancellation; and authentication with multi-user support. Persistence is handled via SQLite (with a custom migration system) or external MySQL/MariaDB. The frontend is built with React, Chakra UI, and nanostores/Redux for state management, with extensive internationalization scaffolding. While primarily an application, the repository contains significant research-oriented code for model architecture definitions, quantization (bitsandbytes, GGUF, SDNQ), custom PyTorch module autocasting, and image processing pipelines (Canny, depth estimation, face detection, segment anything, upscaling). The codebase demonstrates high engineering craftsmanship with abstract service layers, registry patterns, and a node-based architecture that cleanly separates concerns. However, it carries maintenance complexity from two parallel frontend codebases, vendored dependencies, and a large migration chain.

RATING: 9.3/10 MAINT: active DRIFT: low
▸ 4 reusable components found
calesthio/OpenMontage ★ 62522 7980 APPLICATION

OpenMontage is an open-source, agentic video production platform where an AI coding assistant (Claude Code, Cursor) orchestrates the entire pipeline. It provides a library of 100+ tools (video generation, audio, graphics, enhancement, analysis, avatar, subtitles) behind a scored provider selection layer, and 10+ production pipeline manifests (animated explainer, cinematic, documentary montage, talking head, etc.) that define stage order, approval gates, and tool assignments. The core architecture is a Python framework with a BaseTool class, ToolRegistry for auto-discovery, pipeline loader, checkpoint system, and a Remotion-based video composer. It also includes a Backlot living storyboard server for live production monitoring and an Ink Theater procedural puppetry engine. While primarily an application, it contains data/research elements such as ManimGL/ManimCE educational animation examples covering transformer internals, quantum mechanics, and ML concepts, as well as a CLIP-based corpus builder for documentary montage pipelines.

RATING: 8.7/10 MAINT: active DRIFT: medium
▸ 6 reusable components found
LykosAI/StabilityMatrix ★ 8917 609 APPLICATION

Stability Matrix is a cross-platform desktop application (Windows, macOS, Linux) that serves as a unified package manager, model browser, and inference UI for AI image generation tools like Stable Diffusion, ComfyUI, Fooocus, and Forge. The codebase is built with .NET 9 using Avalonia UI (with legacy WPF remnants) and follows the MVVM pattern with CommunityToolkit.Mvvm and DI via Injectio. Key entry points include the MainWindow, PackageManagerPage, InferencePage (text-to-image, image-to-image, video, upscale), and CheckpointBrowserPage (CivitAI, HuggingFace, OpenModelDB). The system manages Python virtual environments, installs packages via pip/uv, and orchestrates inference through ComfyUI's API or built-in Gemini providers. External dependencies include Refit for API clients, LiteDB for local caching, and SkiaSharp for image processing. The architecture is a modular monolith with clear separation of concerns via service interfaces, view models, and custom controls.

RATING: 8.7/10 MAINT: active DRIFT: low
▸ 5 reusable components found
Comfy-Org/ComfyUI ★ 135780 16104 APPLICATION

ComfyUI is a modular, node-based AI engine for content creation, primarily focused on generative image, video, audio, and 3D model inference. It provides a web server (aiohttp) with WebSocket and REST APIs, a visual workflow editor, and an extensible plugin system for custom nodes. The architecture is a monolith with a clear separation of concerns: core model loading and sampling (comfy/), execution engine (comfy_execution/), asset management (app/assets/), external API integrations (comfy_api_nodes/), and a public API for defining nodes (comfy_api/). It supports a vast range of open-source diffusion models (Flux, SD3, Wan, Hunyuan, etc.) and offers advanced features like dynamic VRAM management, quantization, multi-GPU support, and experimental guidance methods. The system operates as a local-first application but also integrates with cloud services via partner API nodes. Key entry points include main.py (startup), server.py (PromptServer), and execution.py (PromptExecutor). The codebase demonstrates high craftsmanship in its weight adapter hierarchy (LoRA, LoHa, OFT), patcher extension system for model hooks, and asset management services with database-backed metadata extraction and cursor-based pagination. However, it also contains experimental and deprecated nodes, some code duplication across model implementations, and a reliance on external dependencies that may drift over time.

RATING: 8.6/10 MAINT: active DRIFT: medium
▸ 5 reusable components found
LuqP2/Image-MetaHub ★ 322 25 APPLICATION

Image MetaHub is a local-first Electron desktop application for organizing, browsing, and re-generating AI-generated images. It indexes media files in place (no uploads, no moving files) and extracts structured metadata — prompts, models, LoRAs, samplers, seeds, workflows — from 15+ generators including ComfyUI, A1111, InvokeAI, Midjourney, DALL-E, Fooocus, and Adobe Firefly (via C2PA/JUMBF/CBOR parsing). Beyond images it handles video, audio, and 3D model metadata. The app provides tagging, collections, automation rules, ratings, analytics, image comparison with pixel-level diffing, lineage/provenance tracking, and integration with local ComfyUI/A1111 instances for re-generation and upscaling. Pro features (visual search, clustering, batch export) are gated by an offline license system with a 7-day trial and Stripe billing.

RATING: 8.5/10 MAINT: active DRIFT: medium
▸ 6 reusable components found
brycedrennan/imaginAIry ★ 8180 471 APPLICATION

The system is a comprehensive toolkit for AI-driven image and video generation, primarily leveraging diffusion models such as Stable Diffusion, Stable Diffusion XL, and Stable Video Diffusion. It provides both a command-line interface (CLI) and an HTTP API server for generating images from text prompts, performing image-to-image translation, inpainting, ControlNet-guided generation, upscaling, face restoration, and video frame interpolation. The architecture is modular, separating core model implementations, enhancements, processors, and utilities, with a strong emphasis on extensibility through configurable backends and pluggable components. Key features include support for multiple sampling algorithms (DDIM, PLMS, K-diffusion solvers), integration with ControlNet for conditional generation via edge, depth, pose, and other maps, and post-processing pipelines for upscaling (using RealESRGAN or similar) and face enhancement (using CodeFormer). The system also incorporates multimodal capabilities through integrated models like BLIP for image captioning and visual question answering, and segment anything model (SAM) for segmentation tasks. Model weight management is handled through a centralized system that resolves and converts weights from various sources (Hugging Face Hub, local cache) and formats (CompVis, Diffusers, Refiners). The CLI offers commands like `imagine` for text-to-image generation, `aimg` as a multi-command interface for imagine, upscale, edit, server, and videogen operations, and `run-server` to launch the HTTP API with a web interface (StableStudio). The HTTP API provides REST endpoints for programmatic access, complete with OpenAPI documentation and GPU-aware concurrency control to handle multiple requests safely. Overall, the system is designed to be user-friendly yet powerful, automating complex workflows like model downloads and caching while providing low-level control for advanced users.

RATING: 8.4/10 MAINT: maintained DRIFT: low
▸ 3 reusable components found
AUTOMATIC1111/stable-diffusion-webui ★ 165127 31637 APPLICATION

Stable Diffusion Web UI is a Gradio-based web application that provides a comprehensive interface for generating and manipulating images using diffusion models. It supports txt2img, img2img, inpainting, outpainting, and various post-processing operations including face restoration and upscaling. The application implements a plugin architecture through extensions, supports multiple model architectures (SD 1.x, SD 2.x, SDXL, SD3), and provides a REST API for programmatic access. The codebase is a monolithic application with tightly coupled modules handling model loading, image processing, UI rendering, and script execution. The architecture centers around a main webui.py entry point that initializes the application, sets up Gradio interfaces, and manages the event loop. The modules directory contains core functionality including sd_models.py for model management, processing.py for the main generation pipeline, and various hijack modules that monkey-patch diffusion model components to add custom behavior like attention optimization and CLIP modification. The extensions-builtin directory contains first-party extensions for LoRA, hypernetworks, and various post-processing operations. Key technical patterns include extensive use of monkey-patching and function hijacking to modify model behavior without changing upstream code, a callback system for script hooks, and a queue-based generation system with progress tracking. The codebase demonstrates sophisticated handling of GPU memory management, model optimization techniques, and cross-platform compatibility. However, the code exhibits significant technical debt with large monolithic files, tight coupling between modules, and complex conditional logic handling multiple model versions and configurations. The repository includes comprehensive test coverage for API endpoints, model utilities, and image processing functions. Dependencies are well-pinned in requirements_versions.txt, though some versions are notably behind current releases (Gradio 3.41.2, Transformers 4.30.2). The codebase shows active maintenance with recent commits and a detailed changelog tracking features and bug fixes across multiple versions.

RATING: 7.8/10 MAINT: active DRIFT: medium
▸ 8 reusable components found
yangjian102621/geekai ★ 4688 1069 APPLICATION

GeekAI is a full-stack, production-grade AI creation platform providing chat, image generation (MidJourney, DALL-E, Jimeng, Stable Diffusion), video generation (Sora, Veo, Keling, MiniMax, Wan, Doubao), music generation (Suno), and PPT creation. The system features a Go + Gin backend with MySQL/Redis persistence and a Vue 3 (Element Plus/Vant) frontend supporting both desktop and mobile interfaces. It includes a complete monetization framework with power credits, multiple payment gateways (Alipay, WeChat Pay, EPay), user invite/reward system, and an administrative dashboard for managing users, models, API keys, and system configurations. Architecturally, the backend follows a layered pattern with handlers, services, and stores. Key design patterns include an adapter pattern for video generation providers (allowing easy addition of new AI services), a factory pattern for OSS uploaders (local, MinIO, Qiniu, Aliyun, Tencent), and an abstraction for payment integrations. Async tasks (e.g., image generation, video creation) are handled via Redis-based queues with polling for status updates. The frontend uses Pinia stores for state management and includes reusable UI components like custom tabs, file uploaders, and parameter builders. The codebase demonstrates solid engineering with moderate code quality. It has several genuinely reusable patterns (video adapter, upload manager, payment service interface) but also exhibits duplication between desktop and mobile views, some hardcoded configuration, and reliance on many third-party SDKs that may drift over time. While not exceptionally polished, it is a fully functional commercial-grade platform suitable for deployment and customization.

RATING: 7.7/10 MAINT: active DRIFT: medium
▸ 4 reusable components found
starik222/BooruDatasetTagManager ★ 1938 155 APPLICATION

BooruDatasetTagManager is a Windows Forms desktop application (C# .NET) for editing tags in image datasets used to train Stable Diffusion hypernetworks, embeddings, and LoRAs. It loads a folder of images with associated .txt tag files, provides a rich UI for tag editing (add, delete, reorder, weight, undo/redo), batch operations, filtering, translation, and auto-tagging via a companion Python AI server (Flask/FastAPI). The Python server supports multiple vision-language models (BLIP, Florence2, JoyCaption, Qwen, etc.) for captioning and tagging, plus background removal and translation. The architecture is a monolithic WinForms client with a separate AI inference server, communicating over HTTP. The codebase is functional and feature-rich but shows signs of monolithic design (large forms, static global state) and some deprecated practices (BinaryFormatter). Several components, such as the universal diffusion metadata parser (Metadata.cs) and the WebP codec wrapper, are well-isolated and potentially reusable in other projects.

RATING: 7.2/10 MAINT: maintained DRIFT: medium
▸ 5 reusable components found
adobe-research/custom-diffusion ★ 1977 140 DATA_RESEARCH

The code answers the research question of how to customize large text-to-image diffusion models with minimal data and minimal parameter updates, and it compares against DreamBooth and Textual Inversion. It includes reusable components for few-shot personalization, delta checkpoint handling, lightweight model compression, multi-concept weight composition, and CLIP/DINO evaluation, although the codebase shows research-level roughness: duplicated CompVis/diffusers paths, hardcoded paths, and several known bugs in conversion and composition utilities.

RATING: 7.1/10 REPRO: 5.0/10 MAINT: active DRIFT: medium
▸ 5 reusable components found
XmYx/deforum-comfy-nodes ★ 197 27 APPLICATION

This repository provides custom nodes for ComfyUI to enable Deforum-based animation generation. It implements frame-by-frame processing of stable diffusion models with parameter scheduling, allowing users to create animated sequences by evolving prompts, seeds, strength, and other parameters over time. The architecture centers around ComfyUI's node system, with specialized nodes for parameter scheduling, prompt engineering, sampling, latent caching, ControlNet integration, image processing, logic control, video I/O, audio synchronization, and advanced noise generation. A critical execution hijack mechanism modifies ComfyUI's internal processing to handle list operations for frame sequences. The system relies on external libraries like Deforum Studio, ControlNet, RAFT, depth models, and FILM for computer vision techniques.

RATING: 7.1/10 MAINT: active DRIFT: medium
▸ 3 reusable components found
RupertAvery/DiffusionToolkit ★ 1012 73 APPLICATION

Diffusion Toolkit is a WPF desktop application for organizing, searching, and browsing AI-generated images. It scans folders for images, extracts generation parameters (prompt, seed, model, CFG, workflow, etc.) from multiple formats (AUTOMATIC1111, ComfyUI, NovelAI, etc.), and stores them in a SQLite database for fast metadata search, filtering, and album management. The application provides a rich thumbnail grid, preview pane, folder/album/tag navigation, and supports videos, external applications integration, and auto‑updating via GitHub releases. Although it demonstrates a depth of domain knowledge and addresses a real need for AI artists, the codebase has notable architectural anti‑patterns (global static service locator, monolithic classes, raw SQL in migrations, large XAML files), is tied to Windows/WPF with no headless component, and uses a forked version of sqlite‑net. The code is actively maintained with frequent commits, but its reusability is limited by tight coupling and lack of an inversion of control container.

RATING: 6.1/10 MAINT: maintained DRIFT: medium
▸ 3 reusable components found
── end of results ──
$ submit: