# ============================================================================
# CivitAI Model Configuration
# ============================================================================
#
# This configuration file defines all available CivitAI models for download.
# Models are organized by category: NSFW checkpoints, LoRAs, embeddings, etc.
#
# Each model entry contains:
#   - name: Filesystem-friendly model name
#   - version_id: CivitAI version ID (used for downloads)
#   - model_id: CivitAI model ID (used for metadata queries)
#   - description: Human-readable description
#   - size_gb: Approximate size in gigabytes
#   - essential: Whether this is an essential model (true/false)
#   - type: Model type (checkpoints, loras, embeddings, etc.)
#   - tags: Array of descriptive tags
#
# ============================================================================

# Global settings
settings:
  cache_dir: /workspace/models/civitai
  output_dir: /workspace/ComfyUI/models
  parallel_downloads: 1
  retry_attempts: 3
  timeout_seconds: 3600
  rate_limit_delay: 5  # seconds between downloads

# Model categories
model_categories:
  # ==========================================================================
  # NSFW IMAGE GENERATION MODELS
  # ==========================================================================
  nsfw_checkpoints:
    - name: lustify-v7-ggwp
      version_id: "1094291"
      model_id: "573152"
      description: "LUSTIFY v7.0 GGWP - Photoreal NSFW checkpoint for women in sexual scenarios"
      size_gb: 6.31
      essential: true
      type: checkpoints
      format: safetensors
      base_model: SDXL 1.0
      vram_gb: 12
      tags: [nsfw, realistic, photography, women]
      recommended_settings:
        sampler: "DPM++ 2M SDE / DPM++ 3M SDE"
        scheduler: "Exponential / Karras"
        steps: 30
        cfg_scale: "4-7"
      notes: "Understands both danbooru tags and natural language prompting"

    - name: pony-diffusion-v6-xl
      version_id: "135867"
      model_id: "257749"
      description: "Pony Diffusion V6 XL - Versatile anime/cartoon/furry model with balanced content"
      size_gb: 6.5
      essential: true
      type: checkpoints
      format: safetensors
      base_model: SDXL 1.0
      vram_gb: 12
      tags: [nsfw, anime, furry, cartoon, versatile]
      training_info:
        images: "2.6M aesthetically ranked"
        ratio: "1:1:1 safe/questionable/explicit"
      notes: "Most popular anime/furry model on CivitAI, supports wide range of styles"

    - name: realvisxl-v5
      version_id: "904692"
      model_id: "139562"
      description: "RealVisXL V5.0 - Photorealistic model for high-quality realistic images"
      size_gb: 6.8
      essential: true
      type: checkpoints
      format: safetensors
      base_model: SDXL 1.0
      vram_gb: 12
      tags: [nsfw, realistic, photorealistic, professional]
      recommended_settings:
        sampler: "DPM++ SDE Karras (30+ steps) or DPM++ 2M Karras (50+ steps)"
      notes: "Excellent for photorealistic portraits and scenes, both SFW and NSFW"

    - name: wai-nsfw-illustrious-sdxl
      version_id: "1239648"
      model_id: "827184"
      description: "WAI-NSFW-illustrious-SDXL v15.0 - Actively updated NSFW Illustrious variant"
      size_gb: 6.5
      essential: false
      type: checkpoints
      format: safetensors
      base_model: SDXL 1.0
      vram_gb: 12
      tags: [nsfw, anime, illustrious, artistic]
      notes: "Latest version from October 2025, frequently updated"

    - name: talmendoxl
      version_id: "131869"
      model_id: "119202"
      description: "TalmendoXL - Uncensored SDXL model biased towards photorealism"
      size_gb: 6.6
      essential: false
      type: checkpoints
      format: safetensors
      base_model: SDXL 1.0
      vram_gb: 12
      tags: [nsfw, realistic, uncensored, photorealistic]
      notes: "Known for high-quality photorealistic NSFW content"

    - name: big-lust-v1-6
      version_id: "649023"
      model_id: "575395"
      description: "Big Lust v1.6 - SDXL NSFW checkpoint"
      size_gb: 6.5
      essential: false
      type: checkpoints
      format: safetensors
      base_model: SDXL 1.0
      vram_gb: 12
      tags: [nsfw, versatile]
      notes: "General-purpose NSFW model with good results across styles"

# ============================================================================
# STORAGE & VRAM SUMMARIES
# ============================================================================

storage_requirements:
  essential_only:
    nsfw_checkpoints: 19.61  # LUSTIFY + Pony + RealVisXL
    total: 19.61

  all_models:
    nsfw_checkpoints: 39.21  # All 6 checkpoints
    total: 39.21

vram_requirements:
  # For 24GB GPU (RTX 4090)
  notes: |
    All SDXL checkpoints require approximately 12GB VRAM.
    Can run one checkpoint at a time on 24GB GPU.
    For simultaneous loading with other models (ComfyUI, vLLM),
    ensure combined VRAM usage stays under 24GB.

# ============================================================================
# INSTALLATION PROFILES
# ============================================================================

installation_profiles:
  essential:
    description: Essential NSFW models only (3 checkpoints)
    categories: [nsfw_checkpoints]
    essential_only: true
    storage_gb: 19.61
    estimated_time: "30-60 minutes"
    models:
      - lustify-v7-ggwp
      - pony-diffusion-v6-xl
      - realvisxl-v5

  complete:
    description: All NSFW models (6 checkpoints)
    categories: [nsfw_checkpoints]
    storage_gb: 39.21
    estimated_time: "1-2 hours"
    models:
      - lustify-v7-ggwp
      - pony-diffusion-v6-xl
      - realvisxl-v5
      - wai-nsfw-illustrious-sdxl
      - talmendoxl
      - big-lust-v1-6

  realistic_only:
    description: Photorealistic NSFW models only
    categories: [nsfw_checkpoints]
    storage_gb: 13.11
    estimated_time: "20-40 minutes"
    models:
      - lustify-v7-ggwp
      - realvisxl-v5

  anime_only:
    description: Anime/cartoon NSFW models only
    categories: [nsfw_checkpoints]
    storage_gb: 13.0
    estimated_time: "20-40 minutes"
    models:
      - pony-diffusion-v6-xl
      - wai-nsfw-illustrious-sdxl

# ============================================================================
# METADATA
# ============================================================================

metadata:
  version: 1.0.0
  last_updated: 2025-11-23
  compatible_with:
    - ComfyUI >= 0.1.0
    - SDXL base models
  maintainer: Valknar
  repository: https://github.com/valknar/runpod
  notes: |
    All models listed are NSFW (adult content) and hosted on CivitAI.
    Requires CIVITAI_API_KEY for downloads.
    Models will be downloaded to cache_dir and symlinked to ComfyUI.