chore: enable juririgged polling for devcontainer support

chore: use container mount path variable instead of hardcoded strings

chore: add proper env file initialization for devcontainer

chore: explicitly create other cache folders

chore remove file so it can be recommitted properly

chore: commit init file with executable mode set

chore: remove unneeded debug launch target

chore: add more recommended extensions

chore: remove unneeded git files

chore: fix python preinstall

chore: update cuda index to match main

chore: drop redundant devcontainer specific extensions list and just use global workspace extensions list.

refactor: move to hatchling for python compiler-backend

chore: oh we actually DO need to specify extensions in the devcontainer.json

refactor: enable UV to handle device-specific torch index resolution.

chore: whoops, fix the named volume name

Author: dsisco11

.devcontainer/.vscode/launch.json

@@ -4,13 +4,6 @@
   // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
   "version": "0.2.0",
   "configurations": [
-    {
-      "name": "InvokeAI :: Chrome",
-      "type": "chrome",
-      "request": "launch",
-      "url": "http://localhost:9090",
-      "webRoot": "${workspaceFolder}/invokeai/frontend/web"
-    },
     {
       "name": "InvokeAI :: Backend",
       "type": "debugpy",

.devcontainer/Dockerfile

@@ -31,6 +31,7 @@ ENV \
     UV_PROJECT_ENVIRONMENT=${UV_PROJECT_ENVIRONMENT}
 
 ENV \
+    COMPUTE_DEVICE=${COMPUTE_DEVICE} \
     PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1 \
     UV_MANAGED_PYTHON=1 \
@@ -39,16 +40,6 @@ ENV \
     INVOKEAI_ROOT=${INVOKEAI_ROOT} \
     HF_HOME=${HF_HOME}
 
-# Set the secondary index for UV to the appropriate PyTorch index URL based on the GPU driver and architecture
-# NOTE: this is the index URL that will be used by `uv` to install PyTorch and other packages
-# Instead of using an envar we write the value to the user-level config file for UV
-RUN if [ "$TARGETPLATFORM" = "linux/arm64" ] || [ "$COMPUTE_DEVICE" = "cpu" ]; then export UV_INDEX="https://download.pytorch.org/whl/cpu"; \
-    elif [ "$COMPUTE_DEVICE" = "rocm" ]; then export UV_INDEX="https://download.pytorch.org/whl/rocm6.2"; \
-    else export UV_INDEX="https://download.pytorch.org/whl/cu124"; \
-    fi \
-    && mkdir -p /home/${USERNAME}/.config/uv/ \
-    && echo "[[index]]\nurl = '$(echo $UV_INDEX)'\n" > /home/${USERNAME}/.config/uv/uv.toml
-
 # Set Python version
 # NOTE: this is the version of Python that will be installed by `uv`
 ARG PYTHON_VERSION=3.12
@@ -60,31 +51,40 @@ UV_PYTHON=${PYTHON_VERSION}
 ARG CACHE_DIR=/home/${USERNAME}/.cache
 # Setting XDG_CACHE_HOME will affect the location of the cache for pnpm and uv, and possibly other tools
 # https://specifications.freedesktop.org/basedir-spec/basedir-spec-latest.html
+# We set XDG_DATA_HOME to a non-volume-mounted location to guarantee that UV pre-installs python. (if its mounted it will only be installed after container start)
 ENV XDG_CACHE_HOME=${CACHE_DIR} \
+    XDG_DATA_HOME=/home/${USERNAME}/.data \
     GIT_LFS_CACHE=${CACHE_DIR}/git-lfs \
     PATH="$UV_PROJECT_ENVIRONMENT/bin:$CACHE_DIR/pnpm:$CACHE_DIR/uv:$PATH" \
     PYTHONPATH="$PYTHONPATH:${INVOKEAI_SRC}"
 
-# Setup cache volume for dependencies
-VOLUME [ "${CACHE_DIR}", "${INVOKEAI_SRC}", "${HF_HOME}"]
+# Predeclare volume mounting points for directories which will be written to.
+# This guarantees that even if the devcontainer doesnt mount a volume for these paths, their contents will still be handled by a volume, 
+# rather than being written to the container image layers itself.
+VOLUME [ "${CACHE_DIR}", "${INVOKEAI_SRC}", "${INVOKEAI_ROOT}", "${HF_HOME}"]
 
 # Create & set ownership of directories
 RUN set -eux; \
     mkdir -p ${CACHE_DIR} && \
     mkdir -p ${CACHE_DIR}/uv && \
     mkdir -p ${CACHE_DIR}/pnpm && \
+    mkdir -p ${CACHE_DIR}/node && \
+    mkdir -p ${CACHE_DIR}/Microsoft && \
     mkdir -p ${INVOKEAI_ROOT}/models/.download_cache && \
     mkdir -p ${UV_PROJECT_ENVIRONMENT} && \
     chown --recursive ${USERNAME}:${USERNAME} ${CACHE_DIR} && \
     chown --recursive ${USERNAME}:${USERNAME} ${INVOKEAI_ROOT} && \
-    chown --recursive ${USERNAME}:${USERNAME} ${UV_PROJECT_ENVIRONMENT} 
+    chown --recursive ${USERNAME}:${USERNAME} ${UV_PROJECT_ENVIRONMENT} && \
+    mkdir -p /home/${USERNAME}/.data && \
+    chown --recursive ${USERNAME}:${USERNAME} /home/${USERNAME}/.data
 
 # Setup PNPM
 RUN corepack use pnpm && corepack enable
 
 # Install `uv` for package management
 COPY --from=ghcr.io/astral-sh/uv:0.6 /uv /uvx /bin/
 
-RUN uv python install ${PYTHON_VERSION}
+RUN uv venv && uv python install ${PYTHON_VERSION}
+ENV UV_PYTHON_DOWNLOADS=never
 
 WORKDIR ${INVOKEAI_SRC}

.devcontainer/devcontainer.json

@@ -7,7 +7,7 @@
       "USERNAME": "node",
       "PYTHON_VERSION": "3.12", // The python version which will be installed in the container
       "INVOKEAI_ROOT": "/home/node/invokeai",
-      "INVOKEAI_SRC": "/workspaces/InvokeAI"
+      "INVOKEAI_SRC": "${containerWorkspaceFolder}"
     }
   },
   "mounts": [
@@ -38,7 +38,8 @@
   ],
   "containerEnv": {
     "UV_COMPILE_BYTECODE": "1",
-    "INVOKEAI_DEV_RELOAD": "true"
+    "INVOKEAI_DEV_RELOAD": "true",
+    "UV_LINK_MODE": "symlink" // Using symlink because they are fast and always work, if you encounter issues after a cache purge, just rebuild the container.
   },
   "features": {
     "ghcr.io/devcontainers/features/git-lfs": {},
@@ -48,19 +49,18 @@
     "gpu": "optional"
   },
   "runArgs": ["--env-file", ".devcontainer/.env"],
+  "initializeCommand": [".devcontainer/scripts/init"],
   "onCreateCommand": "pnpm config --global set store-dir $XDG_CACHE_HOME/pnpm",
-  "updateContentCommand": {
-    "Frontend::Setup": "pnpm install --dir invokeai/frontend/web --reporter append-only --prefer-offline",
-    "Backend::Setup": "uv sync --inexact --all-extras --no-extra onnx-directml"
-  },
+  "updateContentCommand": "pnpm install --dir invokeai/frontend/web --prefer-offline",
+  "postAttachCommand": "uv sync --extra using-$(echo ${COMPUTE_DEVICE:-cpu}) --extra dev --extra test --extra docs",
   "customizations": {
     "vscode": {
       "settings": {
         "python.defaultInterpreterPath": "/home/node/.venv/bin/python",
         "python.analysis.extraPaths": [
-          "/invokeai",
-          "/workspaces/InvokeAI",
-          "/home/node/.venv/lib"
+          "/home/node/invokeai",
+          "/home/node/.venv/lib",
+          "${containerWorkspaceFolder}"
         ],
         "files.autoSaveWhenNoErrors": true,
         "editor.formatOnSave": true,
@@ -70,18 +70,21 @@
         "ruff.linting.enabled": true
       },
       "extensions": [
-        "eamodio.gitlens",
+        "ms-vscode-remote.remote-containers",
+        "ms-azuretools.vscode-docker",
         "ms-toolsai.jupyter",
         "ms-toolsai.vscode-jupyter-cell-tags",
         "ms-toolsai.jupyter-renderers",
         "ms-toolsai.vscode-jupyter-slideshow",
+        "ms-python.python",
         "charliermarsh.ruff",
         "esbenp.prettier-vscode",
         "DavidAnson.vscode-markdownlint",
         "antfu.vite",
         "docker.docker",
-        "ms-vscode-remote.remote-containers",
-        "ms-azuretools.vscode-docker"
+        "dbaeumer.vscode-eslint",
+        "vitest.explorer",
+        "tamasfe.even-better-toml"
       ]
     }
   }

.devcontainer/scripts/init

@@ -0,0 +1 @@
+touch .env

.devcontainer/scripts/init.cmd

@@ -0,0 +1,2 @@
+@ECHO off
+if not exist .env copy NUL .env

.vscode/extensions.json

@@ -1,15 +1,19 @@
 {
-    "recommendations": [
-        "charliermarsh.ruff",
-        "esbenp.prettier-vscode",
-        "DavidAnson.vscode-markdownlint",
-        "antfu.vite",
-        "docker.docker",
-        "ms-vscode-remote.remote-containers",
-        "ms-azuretools.vscode-docker",
-        "ms-toolsai.jupyter",
-        "ms-toolsai.vscode-jupyter-cell-tags",
-        "ms-toolsai.jupyter-renderers",
-        "ms-toolsai.vscode-jupyter-slideshow",
-    ]
-}
+  "recommendations": [
+    "ms-vscode-remote.remote-containers",
+    "ms-azuretools.vscode-docker",
+    "ms-toolsai.jupyter",
+    "ms-toolsai.vscode-jupyter-cell-tags",
+    "ms-toolsai.jupyter-renderers",
+    "ms-toolsai.vscode-jupyter-slideshow",
+    "ms-python.python",
+    "charliermarsh.ruff",
+    "esbenp.prettier-vscode",
+    "DavidAnson.vscode-markdownlint",
+    "antfu.vite",
+    "docker.docker",
+    "dbaeumer.vscode-eslint",
+    "vitest.explorer",
+    "tamasfe.even-better-toml"
+  ]
+}

invokeai/app/invocations/custom_nodes/.gitignore

@@ -57,7 +57,7 @@ def enable_dev_reload(custom_nodes_path=None) -> None:
         paths = [str(invokeai_source_dir() / "*.py")]
         if custom_nodes_path:
             paths.append(str(custom_nodes_path / "*.py"))
-        jurigged.watch(pattern=paths, logger=InvokeAILogger.get_logger(name="jurigged").info)
+        jurigged.watch(pattern=paths, logger=InvokeAILogger.get_logger(name="jurigged").info, poll=True)
 
 
 def apply_monkeypatches() -> None:

invokeai/app/invocations/custom_nodes/.gitinclude

@@ -1,6 +1,6 @@
 [build-system]
-requires = ["setuptools", "pip", "wheel"]
-build-backend = "setuptools.build_meta"
+requires = ["hatchling>=1.25.0"]
+build-backend = "hatchling.build"
 
 [project]
 name = "InvokeAI"
@@ -38,20 +38,19 @@ dependencies = [
   "compel==2.0.2",
   "diffusers[torch]==0.33.0",
   "gguf",
-  "invisible-watermark==0.2.0",           # needed to install SDXL base and refiner using their repo_ids
-  "mediapipe==0.10.14",                   # needed for "mediapipeface" controlnet model
+  "invisible-watermark==0.2.0", # needed to install SDXL base and refiner using their repo_ids
+  "mediapipe==0.10.14", # needed for "mediapipeface" controlnet model
   "numpy<2.0.0",
   "onnx==1.16.1",
   "onnxruntime==1.19.2",
   "opencv-python==4.9.0.80",
   "safetensors",
   "sentencepiece",
   "spandrel",
-  "torch~=2.6.0",                         # torch and related dependencies are loosely pinned, will respect requirement of `diffusers[torch]`
-  "torchsde",                             # diffusers needs this for SDE solvers, but it is not an explicit dep of diffusers
+  "torch~=2.6.0", # torch and related dependencies are loosely pinned, will respect requirement of `diffusers[torch]`
+  "torchsde", # diffusers needs this for SDE solvers, but it is not an explicit dep of diffusers
   "torchvision",
   "transformers",
-
   # Core application dependencies, pinned for reproducible builds.
   "fastapi-events",
   "fastapi",
@@ -60,7 +59,6 @@ dependencies = [
   "pydantic",
   "python-socketio",
   "uvicorn[standard]",
-
   # Auxiliary dependencies, pinned only if necessary.
   "blake3",
   "Deprecated",
@@ -82,7 +80,6 @@ dependencies = [
   "xformers>=0.0.28.post1; sys_platform!='darwin'",
   # torch 2.4+cu carries its own triton dependency
 ]
-
 "onnx" = ["onnxruntime"]
 "onnx-cuda" = ["onnxruntime-gpu"]
 "onnx-directml" = ["onnxruntime-directml"]
@@ -108,6 +105,10 @@ dependencies = [
   "polyfactory==2.19.0",
   "humanize==4.12.1",
 ]
+# Have to declare these here so they are valid values for the `extra` field
+"using-cpu" = []
+"using-cuda" = ["onnxruntime-gpu"]
+"using-rocm" = []
 
 [project.scripts]
 "invokeai-web" = "invokeai.app.run_app:run_app"
@@ -119,37 +120,35 @@ dependencies = [
 "Bug Reports" = "https://github.com/invoke-ai/InvokeAI/issues"
 "Discord" = "https://discord.gg/ZmtBAhwWhy"
 
-[tool.setuptools.dynamic]
-version = { attr = "invokeai.version.__version__" }
+[tool.hatch.version]
+path = "invokeai/version/invokeai_version.py"
+attribute = "__version__"
 
-[tool.setuptools.packages.find]
-"where" = ["."]
-"include" = [
-  "invokeai.assets.fonts*",
-  "invokeai.version*",
-  "invokeai.generator*",
-  "invokeai.backend*",
-  "invokeai.frontend*",
-  "invokeai.frontend.web.dist*",
-  "invokeai.frontend.web.static*",
-  "invokeai.configs*",
-  "invokeai.app*",
-  "invokeai.invocation_api*",
+[tool.hatch.build]
+include = [
+  "invokeai/app/assets/**/*.png",
+  "invokeai/app/services/workflow_records/default_workflows/*.json",
+  "invokeai/app/services/style_preset_records/*.json",
+  "invokeai/app/services/style_preset_images/default_style_preset_images/*.png",
+  "invokeai/assets/fonts/**/*.ttf",
+  "invokeai/backend/**/*.png",
+  "invokeai/configs/*.example",
+  "invokeai/configs/**/*.yaml",
+  "invokeai/configs/*.txt",
+  "invokeai/frontend/web/dist/**",
+  "invokeai/frontend/web/static/**",
+  "invokeai/app/invocations/**",
+  "README.md",
+  "LICENSE",
 ]
+# exclude = ["tests/**", "docs/**"]
+
+[tool.hatch.build.targets.wheel]
+# Wheel gets exactly the same files by default, no extra config needed.
+
+[tool.hatch.build.targets.sdist]
+# Sdist will pull in everything under include/exclude.
 
-[tool.setuptools.package-data]
-"invokeai.app.assets" = ["**/*.png"]
-"invokeai.app.services.workflow_records.default_workflows" = ["*.json"]
-"invokeai.app.services.style_preset_records" = ["*.json"]
-"invokeai.app.services.style_preset_images.default_style_preset_images" = [
-  "*.png",
-]
-"invokeai.assets.fonts" = ["**/*.ttf"]
-"invokeai.backend" = ["**.png", "**/*.icc"]
-"invokeai.configs" = ["*.example", "**/*.yaml", "*.txt"]
-"invokeai.frontend.web.dist" = ["**"]
-"invokeai.frontend.web.static" = ["**"]
-"invokeai.app.invocations" = ["**"]
 
 #=== Begin: PyTest and Coverage
 [tool.pytest.ini_options]
@@ -295,3 +294,37 @@ reportUntypedBaseClass = "warning"
 reportUntypedClassDecorator = "warning"
 reportUntypedFunctionDecorator = "warning"
 reportUntypedNamedTuple = "warning"
+
+#=== Begin: UV
+
+[tool.uv.sources]
+torch = [
+  # { index = "default", marker = "sys_platform == 'darwin'" }, # macOS
+  { index = "pytorch-arm", marker = "sys_platform != 'darwin' and platform_machine == 'arm64'" },
+  # Linux and Windows
+  { index = "pytorch-cu126", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra == 'using-cuda' and extra != 'using-rocm'" },
+  { index = "pytorch-rocm", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra != 'using-cuda' and extra == 'using-rocm'" },
+  { index = "pytorch-cpu", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra != 'using-cuda' and extra != 'using-rocm'" },
+]
+
+[[tool.uv.index]]
+name     = "pytorch-rocm"
+url      = "https://download.pytorch.org/whl/rocm6.2"
+explicit = true
+
+[[tool.uv.index]]
+name     = "pytorch-cu126"
+url      = "https://download.pytorch.org/whl/cu126"
+explicit = true
+
+[[tool.uv.index]]
+name     = "pytorch-arm"
+url      = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[[tool.uv.index]]
+name     = "pytorch-cpu"
+url      = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+#=== End: UV