Merge pull request #24 from KeesGeerligs/kees-dev

More templates

templates/AUTOMATIC1111-stable-diffusion-1.5/README.md

@@ -1,10 +1,20 @@
-# Stable Diffusion WebUI
+# AUTOMATIC1111 Stable Diffusion
 
-![Stable Diffusion WebUI](https://raw.githubusercontent.com/nosana-ci/templates/refs/heads/main/templates/AUTOMATIC1111-stable-diffusion/stable_diff.gif)
+![AUTOMATIC1111](https://raw.githubusercontent.com/nosana-ci/templates/refs/heads/main/templates/AUTOMATIC1111-stable-diffusion/stable_diff.gif)
 
-[Stable Diffusion WebUI](https://github.com/AUTOMATIC1111/stable-diffusion-webui) is a web interface for Stable Diffusion, implemented using Gradio library.
+A powerful web interface for Stable Diffusion, offering extensive features for AI image generation and editing.
 
-Unleash your creativity with Nosana! Effortlessly run a Stable Diffusion instance to generate stunning images.
-Experience the power of advanced AI and GPU-backed nodes, making your image creation process smooth and efficient.
-Whether for personal projects or professional work, Nosana provides the tools you need to bring your artistic visions to life.
+Unleash the power of Stable Diffusion with Nosana! Effortlessly run your instance on high-performance GPU-backed nodes, ensuring optimal image generation for your creative projects.
+
+## Key Features
+- Text-to-image generation
+- Image-to-image editing
+- Inpainting and outpainting
+- Custom model support
+- GPU acceleration support
+
+## Configuration
+- Port: 7860
+- GPU: Required
+- Model: Stable Diffusion 1.5
 

templates/AUTOMATIC1111-stable-diffusion-1.5/info.json

@@ -1,7 +1,7 @@
 {
-  "id": "stable-diffusion",
-  "name": "Stable Diffusion",
+  "id": "AUTOMTIC1111-stable-diffusion",
+  "name": "AUTOMATIC1111 Stable Diffusion 1.5",
   "description": "Stable Diffusion is a latent text-to-image diffusion model",
-  "category": ["AI", "Image Generation", "GPU", "Service", "Featured"],
-  "icon": "https://res.cloudinary.com/dsht8f3ff/image/upload/v1714833633/stabilityai_fiujad.png"
+  "category": ["Web UI", "Image Generation"],
+  "icon": "https://private-user-images.githubusercontent.com/173618923/361423663-9ebdd0cf-02e2-497d-b87b-f98555396d23.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MzE1MTEwOTUsIm5iZiI6MTczMTUxMDc5NSwicGF0aCI6Ii8xNzM2MTg5MjMvMzYxNDIzNjYzLTllYmRkMGNmLTAyZTItNDk3ZC1iODdiLWY5ODU1NTM5NmQyMy5wbmc_WC1BbXotQWxnb3JpdGhtPUFXUzQtSE1BQy1TSEEyNTYmWC1BbXotQ3JlZGVudGlhbD1BS0lBVkNPRFlMU0E1M1BRSzRaQSUyRjIwMjQxMTEzJTJGdXMtZWFzdC0xJTJGczMlMkZhd3M0X3JlcXVlc3QmWC1BbXotRGF0ZT0yMDI0MTExM1QxNTEzMTVaJlgtQW16LUV4cGlyZXM9MzAwJlgtQW16LVNpZ25hdHVyZT1kNDI5MzVkM2I4NGY1NDViNzEzZDE5ZjYxNzJmYjdmOTM0MWU0OGI4OTg4OTgyZGY3NTUyZGQwODIwNzQzOGU2JlgtQW16LVNpZ25lZEhlYWRlcnM9aG9zdCJ9.o4AIHOfkVEEw-Su4HTmAQIzBvnxjcOB1BUiUDzUSZrs"
 }

templates/AUTOMATIC1111-stable-diffusion-1.5/job-definition.json

@@ -11,7 +11,7 @@
       "args": {
         "cmd": [
           "/bin/sh", "-c", 
-          "python -u launch.py --listen --port 7860 --api"
+          "python -u launch.py --listen --port 7860 --enable-insecure-extension-access"
         ],
         "image": "docker.io/nosana/automatic1111:0.0.1",
         "gpu": true,

templates/Axolotl/README.md

@@ -1,9 +1,19 @@
-# Axolotl
+# Axolotl LLM Fine-tuning
 
-![Axolotl](https://raw.githubusercontent.com/nosana-ci/templates/refs/heads/main/templates/axolotl/axolotl.mp4)
+![Axolotl](https://raw.githubusercontent.com/nosana-ci/templates/refs/heads/main/templates/Axolotl/axolotl.mp4)
 
-Axolotl is a powerful tool designed to streamline the fine-tuning of various AI models, offering support for multiple configurations and architectures. It enables efficient training with single or multiple GPUs, making it ideal for both personal and professional projects.
+A streamlined environment for fine-tuning Large Language Models with multiple training configurations.
 
-Unleash your potential with Nosana! Effortlessly run an Axolotl instance to fine-tune your AI models with ease. Experience the flexibility and power of advanced AI and GPU-backed nodes, ensuring your model training processes are smooth and effective.
+Unleash the power of model fine-tuning with Nosana! Effortlessly run your Axolotl environment on high-performance GPU-backed nodes, ensuring optimal training for your custom models.
 
-Whether you're conducting research, developing applications, or exploring AI capabilities, Nosana provides the essential tools to leverage Axolotl's full potential.
+## Key Features
+- Multiple training methods
+- Configuration flexibility
+- Experiment tracking
+- Jupyter interface
+- GPU acceleration support
+
+## Configuration
+- Port: 8888
+- GPU: Required
+- No authentication required

templates/Axolotl/info.json

@@ -1,7 +1,7 @@
 {
   "id": "axolotl",
-  "name": "Axolotl",
+  "name": "Axolotl finetuning LLM's",
   "description": "Axolotl streamlines the fine-tuning of various AI models, supporting multiple configurations and architectures for efficient and flexible model training.",
-  "category": ["AI", "Model Fine-Tuning", "GPU", "Service", "Featured"],
-  "icon": "https://avatars.githubusercontent.com/u/132000632?s=280&v=4"
+  "category": ["Model Fine-Tuning"],
+  "icon": "https://raw.githubusercontent.com/nosana-ci/templates/refs/heads/main/templates/Axolotl/axolotl-nobackground.png"
 }

templates/ComfyUI/README.md

@@ -0,0 +1,19 @@
+# ComfyUI
+
+![ComfyUI](https://raw.githubusercontent.com/nosana-ci/templates/main/templates/ComfyUI/comfyui.gif)
+
+A node-based interface for image generation with advanced workflow capabilities.
+
+Unleash the power of visual AI workflows with Nosana! Effortlessly run your ComfyUI instance on high-performance GPU-backed nodes, ensuring optimal performance for your image generation pipelines.
+
+## Key Features
+- Node-based workflow
+- Custom node creation
+- Pipeline automation
+- Advanced queue system
+- GPU acceleration support
+
+## Configuration
+- Port: 8188
+- GPU: Required
+- Model: Stable Diffusion

templates/ComfyUI/info.json

@@ -0,0 +1,7 @@
+{
+  "id": "comfyui",
+  "name": "ComfyUI",
+  "description": "A powerful and modular image generation interface with a node-based workflow editor",
+  "category": ["Web UI", "Image Generation"],
+  "icon": "https://raw.githubusercontent.com/comfyanonymous/ComfyUI/master/web/favicon.ico"
+}

templates/ComfyUI/job-definition.json

@@ -0,0 +1,29 @@
+{
+  "version": "0.1",
+  "type": "container",
+  "meta": {
+    "trigger": "benchmark"
+  },
+  "ops": [
+    {
+      "type": "container/run",
+      "id": "comfy_stablediffusion",
+      "args": {
+        "cmd": [
+          "/bin/sh", "-c", 
+          "python -u main.py --listen --port 7860 --gpu-only"
+        ],
+        "image": "docker.io/nosana/sd-comfy-manager:0.0.1",
+        "gpu": true,
+        "expose": 7860,
+        "resources": [
+          {
+            "type": "S3",
+            "url": "https://models.nosana.io/stable-diffusion/1.5",
+            "target": "/comfyui/models/checkpoints"
+          }
+        ]
+      }
+    }
+  ]
+}

templates/Forge-stable-diffusion-1.5/README.md

@@ -0,0 +1,19 @@
+# Forge Stable Diffusion
+
+![Forge](https://raw.githubusercontent.com/nosana-ci/templates/main/templates/Forge-Stable-Diffusion/forge.gif)
+
+An optimized fork of AUTOMATIC1111's Stable Diffusion web UI with enhanced features.
+
+Unleash the power of optimized image generation with Nosana! Effortlessly run your Forge instance on high-performance GPU-backed nodes, ensuring optimal performance for your creative projects.
+
+## Key Features
+- Enhanced performance
+- Advanced UI features
+- Custom model support
+- API integration
+- GPU acceleration support
+
+## Configuration
+- Port: 7861
+- GPU: Required
+- Model: Stable Diffusion 1.5

templates/Forge-stable-diffusion-1.5/info.json

@@ -0,0 +1,7 @@
+{
+  "id": "forge-stable-diffusion",
+  "name": "Forge Stable Diffusion",
+  "description": "A fork of AUTOMATIC1111's Stable Diffusion web UI with additional features and optimizations",
+  "category": ["Web UI", "Image Generation"],
+  "icon": "https://raw.githubusercontent.com/lllyasviel/stable-diffusion-webui-forge/master/favicon.ico"
+}

templates/Forge-stable-diffusion-1.5/job-definition.json

@@ -0,0 +1,30 @@
+{
+  "version": "0.1",
+  "type": "container",
+  "meta": {
+    "trigger": "cli"
+  },
+  "ops": [
+    {
+      "type": "container/run",
+      "id": "forge_stablediffusion",
+      "args": {
+        "cmd": [
+          "/bin/sh", "-c", 
+          "python -u launch.py --api --listen --port 7861"
+        ],
+        "image": "docker.io/nosana/sd-forge-bench:1.0.0",
+        "gpu": true,
+        "expose": 7861,
+        "resources": [
+          {
+            "type": "S3",
+            "url": "https://models.nosana.io/stable-diffusion/1.5",
+            "target": "/opt/stable-diffusion-webui-forge/models/Stable-diffusion"
+          }
+        ]
+
+      }
+    }
+  ]
+}

templates/InvokeAI/README.md

@@ -2,6 +2,18 @@
 
 ![Invoke AI](https://raw.githubusercontent.com/nosana-ci/templates/refs/heads/main/templates/InvokeAI/invoke_ai.gif)
 
-Invoke is a leading creative engine built to empower professionals and enthusiasts alike. Generate and create stunning visual media using the latest AI-driven technologies. Invoke offers an industry-leading web-based UI and serves as the foundation for multiple commercial products.
+A powerful creative engine for AI-driven image generation and editing.
 
-Unleash your creativity with Nosana! Effortlessly run an Invoke AI instance to generate stunning images. Experience the power of advanced AI and GPU-backed nodes, making your image creation process smooth and efficient. Whether for personal projects or professional work, Nosana provides the tools you need to bring your artistic visions to life.
+Unleash the power of creative AI with Nosana! Effortlessly run your Invoke AI instance on high-performance GPU-backed nodes, ensuring optimal performance for your artistic projects.
+
+## Key Features
+- Text-to-image generation
+- Image editing tools
+- Workflow automation
+- Custom model support
+- GPU acceleration support
+
+## Configuration
+- Port: 7860
+- GPU: Required
+- Model: Stable Diffusion

templates/InvokeAI/info.json

@@ -2,6 +2,6 @@
   "id": "invoke-ai",
   "name": "Invoke AI",
   "description": "Invoke is a leading creative engine built to empower professionals and enthusiasts alike. Generate and create stunning visual media using the latest AI-driven technologies.",
-  "category": ["AI", "Image Generation", "GPU", "Service", "Featured"],
-  "icon": "https://cdn.prod.website-files.com/6569f44a2a0eba8a49bc11a1/65772970ccfba3673b7db46a_logo-invoke.svg"
+  "category": ["Web UI", "Image Generation"],
+  "icon": "https://d4.alternativeto.net/xsSoWnh09YOroqDfDTYx-QcFdoXjOMcxLtDc7FZleA0/rs:fit:280:280:0/g:ce:0:0/exar:1/YWJzOi8vZGlzdC9pY29ucy9pbnZva2VhaV8yMjc0ODEucG5n.png"
 }

templates/Kohya-SS/README.md

@@ -0,0 +1,19 @@
+# Kohya SS GUI
+
+![Kohya SS](https://raw.githubusercontent.com/nosana-ci/templates/main/templates/Kohya-ss/kohya-ss.gif)
+
+A user-friendly interface for training and fine-tuning Stable Diffusion models.
+
+Unleash the power of model training with Nosana! Effortlessly run your Kohya SS instance on high-performance GPU-backed nodes, ensuring optimal training for your custom models.
+
+## Key Features
+- Custom model training
+- LoRA support
+- Hyperparameter optimization
+- Dataset management
+- GPU acceleration support
+
+## Configuration
+- Port: 7860
+- GPU: Required
+- Headless mode enabled

templates/Kohya-SS/info.json

@@ -0,0 +1,7 @@
+{
+  "id": "kohya-ss",
+  "name": "Kohya SS GUI",
+  "description": "A user-friendly web interface for training and fine-tuning Stable Diffusion models using Kohya SS scripts.",
+  "category": ["Web UI", "Image Generation Fine-tuning"],
+  "icon": "https://raw.githubusercontent.com/nosana-ci/templates/main/templates/Kohya-ss/kohya-ss.png"
+}

templates/Kohya-SS/job-definition.json

@@ -0,0 +1,23 @@
+{
+  "version": "0.1",
+  "type": "container",
+  "meta": {
+    "trigger": "cli"
+  },
+  "ops": [
+    {
+      "type": "container/run",
+      "id": "kohya-ss-webservice",
+      "args": {
+        "image": "nosana/kohya_ss:1.0.0",
+        "cmd": ["python3", "kohya_gui.py", "--listen", "0.0.0.0", "--server_port", "7860", "--headless"],
+        "gpu": true,
+        "expose": 7860,
+        "env": {
+          "NVIDIA_VISIBLE_DEVICES": "all",
+          "NVIDIA_DRIVER_CAPABILITIES": "compute,utility"
+        }
+      }
+    }
+  ]
+}

templates/LMDeploy-API/README.md

@@ -0,0 +1,19 @@
+# LMDeploy API
+
+![LMDeploy](link-to-image)
+
+A high-performance inference engine for Large Language Models with advanced optimization features.
+
+Unleash the power of LLMs with Nosana! Effortlessly deploy your models on high-performance GPU-backed nodes, ensuring optimal inference speed for your applications.
+
+## Key Features
+- Optimized model inference
+- Quantization support
+- RESTful API interface
+- Multi-model serving
+- GPU acceleration support
+
+## Configuration
+- Port: 23333
+- GPU: Required
+- Model: Qwen2.5-7B

templates/LMDeploy-API/info.json

@@ -0,0 +1,7 @@
+{
+  "id": "lmdeploy-api",
+  "name": "LMDeploy API",
+  "description": "A high-performance inference engine for LLMs with quantization and optimization features",
+  "category": ["API Only", "LLM"],
+  "icon": "https://raw.githubusercontent.com/InternLM/lmdeploy/main/docs/en/_static/image/lmdeploy-logo.png"
+} 

templates/LMDeploy-API/job-definition.json

@@ -0,0 +1,23 @@
+{
+  "version": "0.1",
+  "type": "container",
+  "meta": {
+    "trigger": "benchmark"
+  },
+  "ops": [
+    {
+      "type": "container/run",
+      "id": "lmdepoy",
+      "args": {
+        "cmd": [
+          "/bin/sh",
+          "-c",
+          "lmdeploy serve api_server Qwen/Qwen2.5-7B --model-name Qwen2.5-7B"
+        ],
+        "image": "docker.io/openmmlab/lmdeploy:latest",
+        "gpu": true,
+        "expose": 23333
+      }
+    }
+  ]
+} 

templates/Llama-Factory/README.md

@@ -0,0 +1,19 @@
+# Llama Factory
+
+![Llama Factory](link-to-image)
+
+A unified framework for fine-tuning Large Language Models with various training methods.
+
+Unleash the power of model fine-tuning with Nosana! Effortlessly run your Llama Factory instance on high-performance GPU-backed nodes, ensuring optimal training for your models.
+
+## Key Features
+- Multiple training methods
+- Pre-built training templates
+- Model architecture support
+- Progress monitoring
+- GPU acceleration support
+
+## Configuration
+- Port: 7860
+- GPU: Required
+- WebUI access for training

templates/Llama-Factory/info.json

@@ -0,0 +1,7 @@
+{
+  "id": "llama-factory",
+  "name": "Llama Factory",
+  "description": "A unified framework for fine-tuning LLMs, supporting various architectures and training methods",
+  "category": ["Web UI", "LLM Fine-tuning"],
+  "icon": "https://raw.githubusercontent.com/hiyouga/LLaMA-Factory/main/assets/logo.png"
+} 

templates/Llama-Factory/job-definition.json

@@ -0,0 +1,23 @@
+{
+  "version": "0.1",
+  "type": "container",
+  "meta": {
+    "trigger": "cli"
+  },
+  "ops": [
+    {
+      "type": "container/run",
+      "id": "llama-factory",
+      "args": {
+        "image": "docker.io/nosana/llama-factory:0.0.0",
+        "cmd": [
+          "/bin/sh",
+          "-c",
+          "llamafactory-cli webui"
+        ],
+        "gpu": true,
+        "expose": 7860
+      }
+    }
+  ]
+}