It works!

flake.lock

@@ -0,0 +1,79 @@
+{
+  "nodes": {
+    "comfyuigit": {
+      "flake": false,
+      "locked": {
+        "lastModified": 1755071198,
+        "narHash": "sha256-vIw22ISbjUnfRB6+TFE7QKbVnEXu6BFAN8lmCGE74/M=",
+        "owner": "comfyanonymous",
+        "repo": "ComfyUi",
+        "rev": "d5c1954d5cd4a789bbf84d2b75a955a5a3f93de8",
+        "type": "github"
+      },
+      "original": {
+        "owner": "comfyanonymous",
+        "ref": "v0.3.50",
+        "repo": "ComfyUi",
+        "type": "github"
+      }
+    },
+    "flake-utils": {
+      "inputs": {
+        "systems": "systems"
+      },
+      "locked": {
+        "lastModified": 1731533236,
+        "narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1756125398,
+        "narHash": "sha256-XexyKZpf46cMiO5Vbj+dWSAXOnr285GHsMch8FBoHbc=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "3b9f00d7a7bf68acd4c4abb9d43695afb04e03a5",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "comfyuigit": "comfyuigit",
+        "flake-utils": "flake-utils",
+        "nixpkgs": "nixpkgs"
+      }
+    },
+    "systems": {
+      "locked": {
+        "lastModified": 1681028828,
+        "narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
+        "owner": "nix-systems",
+        "repo": "default",
+        "rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-systems",
+        "repo": "default",
+        "type": "github"
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}

flake.nix

@@ -0,0 +1,156 @@
+{
+  description = "ComfyUI as a Nix Flake";
+
+  inputs = {
+     nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
+     flake-utils.url = "github:numtide/flake-utils";
+     comfyuigit = {
+       flake = false;
+       url = "github:comfyanonymous/ComfyUi/v0.3.50";
+     };
+  };
+
+  outputs = { flake-utils, nixpkgs, ... }:
+    flake-utils.lib.eachDefaultSystem (system:
+      let pkgs = import nixpkgs { inherit system; config.rocmSupport = true; };
+          python = (pkgs.python3.withPackages (
+            ps: with ps; [
+              alembic
+              torch
+              torchsde
+              torchvision
+              torchaudio
+              numpy
+              einops
+              transformers
+              tokenizers
+              sentencepiece
+              safetensors
+              aiohttp
+              yarl
+              pyyaml
+              pydantic
+              pydantic-settings
+              pillow
+              scipy
+              tqdm
+              psutil
+              av
+              sqlalchemy
+              soundfile
+            ]
+          ));
+      in rec {
+       
+        packages.comfyui_frontend = pkgs.python3Packages.buildPythonPackage rec {
+          pname = "comfyui-frontend-package";
+          version = "1.23.4";
+          pyproject = true;
+
+          src = pkgs.fetchPypi {
+            pname = "comfyui_frontend_package";
+            inherit version;
+            sha256 = "sha256-8HrnNEO5rOuUez1MEzWUOJ512nraRTfH+BRJD5PwMfg=";
+          };
+
+          buildInputs = [
+            python
+            python.pkgs.pip
+          ];
+        };
+
+        packages.comfyui_workflow = pkgs.python3Packages.buildPythonPackage rec {
+          pname = "comfyui-workflow-templates";
+          version = "0.1.59";
+          pyproject = true;
+
+          src = pkgs.fetchPypi {
+            pname = "comfyui_workflow_templates";
+            inherit version;
+            sha256 = "sha256-dt8IbxOzNIZi36eAjVjcoPIXwaVkLjI2VUixIoIWXBo=";
+          };
+
+          buildInputs = [
+            python
+            python.pkgs.pip
+          ];
+        };
+
+        packages.comfyui_embedded_docs = pkgs.python3Packages.buildPythonPackage rec {
+          pname = "comfyui-embedded-docs";
+          version = "0.2.6";
+          pyproject = true;
+
+          src = pkgs.fetchPypi {
+            pname = "comfyui_embedded_docs";
+            inherit version;
+            sha256 = "sha256-ild/PuIWvo3NbAjpZYxvJX/7np6B9A8NNt6bSZJJdWo=";
+          };
+
+          buildInputs = [
+            python
+            python.pkgs.pip
+          ];
+        };
+
+        packages.comfyui =
+          let python_extended = (python.withPackages (ps: [
+            ps.alembic
+            ps.torchWithRocm
+            ps.torchsde
+            ps.torchvision
+            ps.torchaudio
+            ps.numpy
+            ps.einops
+            ps.transformers
+            ps.tokenizers
+            ps.sentencepiece
+            ps.safetensors
+            ps.pydantic
+            ps.pydantic-settings
+            ps.aiohttp
+            ps.yarl
+            ps.pyyaml
+            ps.pillow
+            ps.scipy
+            ps.tqdm
+            ps.psutil
+            ps.av
+            ps.sqlalchemy
+            ps.soundfile
+            packages.comfyui_frontend
+            packages.comfyui_workflow
+            packages.comfyui_embedded_docs
+          ]));
+          in pkgs.stdenv.mkDerivation {
+          pname = "ComfyUI";
+          version = "0.3.50";
+          pyproject = true;
+
+          src = pkgs.fetchFromGitHub {
+            owner = "comfyanonymous";
+            repo = "ComfyUI";
+            tag = "v0.3.50";
+            sha256 = "sha256-vIw22ISbjUnfRB6+TFE7QKbVnEXu6BFAN8lmCGE74/M=";
+          };
+
+          buildInputs = [
+            python_extended
+            python.pkgs.pip
+            packages.comfyui_frontend
+          ];
+
+          installPhase = ''
+            mkdir -p $out/bin
+            cp -r --no-preserve=mode,ownership $src/* $out
+            chmod -R u+w $out
+
+            echo "#!/bin/sh" > $out/bin/ComfyUI
+            echo "export MALLOC=system" >> $out/bin/ComfyUI
+            echo "exec ${python_extended}/bin/python $out/main.py \"\$@\"" >> $out/bin/ComfyUI
+            chmod +x $out/bin/ComfyUI
+          '';
+        };
+      }
+    );
+}