Merge pull request #16 from outis1one/claude/fix-invokeai-upload-To8AX

Add LoRA import script and InvokeAI usage guide

README.md

@@ -65,6 +65,7 @@ sudo systemctl status local-ai
 | `ubuntu-post-install.sh` | 8,889 | Full Ubuntu 24.04 post-install (dev tools, fonts, apps, tweaks). Run once on a fresh OS install. Independent of the AI stack. |
 | `configure-storage.sh` | 239 | Storage/mount configuration helper. Run separately if you have a secondary drive for AI data. |
 | `kiwix_download.sh` | 198 | Downloads ZIM files (Wikipedia, Stack Overflow, etc.) for offline use. Run separately — files are large. |
+| `invokeai-import-lora.sh` | 85 | Copies a LoRA `.safetensors` file into InvokeAI's Docker model volume. |
 
 ### Which setup script should I use?
 
@@ -131,6 +132,38 @@ Both scripts are **idempotent** — safe to re-run for updates. Config files are
 
 ---
 
+## Using LoRA Models in InvokeAI
+
+LoRA (Low-Rank Adaptation) files let you customize image generation with fine-tuned styles or characters. If you trained a LoRA on RunPod or elsewhere, here's how to use it.
+
+### Import a LoRA file
+
+```bash
+# Copy your LoRA into the InvokeAI Docker volume:
+./invokeai-import-lora.sh ~/Downloads/my-lora.safetensors
+
+# Optionally give it a display name:
+./invokeai-import-lora.sh ~/Downloads/my-lora.safetensors "My Custom Style"
+```
+
+### Use the LoRA in InvokeAI
+
+1. Open InvokeAI at `http://<ip>:9090`
+2. Go to **Model Manager** (cube icon, left sidebar) and click **Scan for Models** / **Sync Models**
+3. Your LoRA should appear in the model list
+4. Switch to **Text to Image** tab
+5. In the left panel, find the **LoRA** section (below the model selector)
+6. Click **+** to add your LoRA, then adjust the **weight** slider (start at 0.7–0.85)
+
+### Troubleshooting greyed-out upload buttons
+
+- **No base model installed:** You need a fully downloaded base model (e.g., SD 1.5) before InvokeAI enables LoRA uploads. Use Model Manager to install one first.
+- **Model not synced:** After copying files, click **Scan for Models** in Model Manager.
+- **Architecture mismatch:** A LoRA trained on SD 1.5 only works with SD 1.5 base models — not SDXL or SD 2.x.
+- **Use the import script instead:** The greyed-out UI upload can be bypassed entirely by using `invokeai-import-lora.sh` to copy files directly into the model volume.
+
+---
+
 ## Updating
 
 Re-run the setup script — it detects an existing install and skips prereqs:

invokeai-import-lora.sh

@@ -0,0 +1,81 @@
+#!/usr/bin/env bash
+# Import a LoRA (.safetensors) file into InvokeAI's Docker volume
+# Usage: ./invokeai-import-lora.sh /path/to/my-lora.safetensors
+set -euo pipefail
+
+RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'
+CYAN='\033[0;36m'; BOLD='\033[1m'; NC='\033[0m'
+
+if [[ $# -lt 1 ]]; then
+    echo -e "${BOLD}Usage:${NC} $0 <lora-file.safetensors> [display-name]"
+    echo ""
+    echo "  Copies a LoRA file into InvokeAI's model volume so it appears"
+    echo "  in the Model Manager automatically."
+    echo ""
+    echo "  Examples:"
+    echo "    $0 ~/Downloads/my-character-lora.safetensors"
+    echo "    $0 ~/Downloads/my-character-lora.safetensors \"My Character\""
+    exit 1
+fi
+
+LORA_FILE="$1"
+DISPLAY_NAME="${2:-$(basename "$LORA_FILE" .safetensors)}"
+
+# Validate file exists
+if [[ ! -f "$LORA_FILE" ]]; then
+    echo -e "${RED}Error:${NC} File not found: $LORA_FILE"
+    exit 1
+fi
+
+# Validate file extension
+if [[ "$LORA_FILE" != *.safetensors && "$LORA_FILE" != *.ckpt && "$LORA_FILE" != *.pt ]]; then
+    echo -e "${YELLOW}Warning:${NC} File doesn't have a typical LoRA extension (.safetensors, .ckpt, .pt)"
+    read -rp "Continue anyway? [y/N] " yn
+    [[ "$yn" != [yY]* ]] && exit 1
+fi
+
+# Check if InvokeAI container exists
+if ! docker ps -a --format '{{.Names}}' 2>/dev/null | grep -q '^invokeai$'; then
+    echo -e "${RED}Error:${NC} InvokeAI container not found. Is the AI stack running?"
+    echo "  Try: bash ~/docker/ai-stack/start.sh"
+    exit 1
+fi
+
+# Check if container is running
+if ! docker ps --format '{{.Names}}' 2>/dev/null | grep -q '^invokeai$'; then
+    echo -e "${YELLOW}InvokeAI container is stopped. Starting it...${NC}"
+    docker start invokeai
+    sleep 3
+fi
+
+FILENAME=$(basename "$LORA_FILE")
+
+echo -e "${CYAN}[..]${NC} Copying ${BOLD}$FILENAME${NC} into InvokeAI..."
+
+# Copy the LoRA file into the container's models directory
+# InvokeAI looks for LoRA files in /invokeai/models/lora/
+docker exec invokeai mkdir -p /invokeai/models/lora
+docker cp "$LORA_FILE" "invokeai:/invokeai/models/lora/$FILENAME"
+
+echo -e "${GREEN}[OK]${NC} LoRA file copied successfully!"
+echo ""
+echo -e "${BOLD}Next steps in InvokeAI (http://localhost:9090):${NC}"
+echo ""
+echo "  1. Open the ${BOLD}Model Manager${NC} (cube icon in the left sidebar)"
+echo "  2. Click ${BOLD}\"Scan for Models\"${NC} or ${BOLD}\"Sync Models\"${NC} button"
+echo "     - Your LoRA '${DISPLAY_NAME}' should appear in the list"
+echo "  3. If it doesn't auto-detect, click ${BOLD}\"Add Model\" > \"Scan Folder\"${NC}"
+echo "     and enter: ${BOLD}/invokeai/models/lora${NC}"
+echo ""
+echo -e "${BOLD}To use the LoRA when generating images:${NC}"
+echo ""
+echo "  1. Go to the ${BOLD}Text to Image${NC} or ${BOLD}Image to Image${NC} tab"
+echo "  2. In the left panel, find the ${BOLD}\"LoRA\"${NC} section"
+echo "     (expand it if collapsed — it's below the main model selector)"
+echo "  3. Click ${BOLD}\"+\"${NC} to add your LoRA from the dropdown"
+echo "  4. Adjust the ${BOLD}weight${NC} slider (start with 0.7–0.85)"
+echo "  5. Make sure your ${BOLD}base model${NC} matches what the LoRA was trained on"
+echo "     (e.g., if trained on SD 1.5, select a SD 1.5 checkpoint)"
+echo ""
+echo -e "${YELLOW}Tip:${NC} If the LoRA was trained on SD 1.5, you MUST use an SD 1.5"
+echo "     base model — it won't work with SDXL or other architectures."