acw/comfyui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

It works!

Browse Source
This commit is contained in:
Adam Wick
2025-09-04 20:43:14 -07:00
parent b8f8e4d6a6
commit 94328d682d
2 changed files with 137 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
flake.lock generated
View File

@@ -37,11 +37,11 @@
}, },
"nixpkgs": { "nixpkgs": {
"locked": { "locked": {
"lastModified": 1755186698, "lastModified": 1756125398,
"narHash": "sha256-wNO3+Ks2jZJ4nTHMuks+cxAiVBGNuEBXsT29Bz6HASo=", "narHash": "sha256-XexyKZpf46cMiO5Vbj+dWSAXOnr285GHsMch8FBoHbc=",
"owner": "nixos", "owner": "nixos",
"repo": "nixpkgs", "repo": "nixpkgs",
"rev": "fbcf476f790d8a217c3eab4e12033dc4a0f6d23c", "rev": "3b9f00d7a7bf68acd4c4abb9d43695afb04e03a5",
"type": "github" "type": "github"
}, },
"original": { "original": {

179
flake.nix
View File

@@ -10,58 +10,147 @@
}; };
}; };
outputs = { comfyuigit, flake-utils, nixpkgs, ... }: outputs = { flake-utils, nixpkgs, ... }:
flake-utils.lib.eachDefaultSystem (system: flake-utils.lib.eachDefaultSystem (system:
let pkgs = import nixpkgs { let pkgs = import nixpkgs { inherit system; config.rocmSupport = true; };
inherit system; python = (pkgs.python3.withPackages (
}; ps: with ps; [
pyenv = pkgs.python312.buildEnv.override { alembic
extraLibs = with pkgs.python312Packages; [ torch
torchvision torchsde
torchaudio torchvision
torchsde torchaudio
numpy
einops einops
transformers transformers
safetensors tokenizers
pyyaml sentencepiece
pillow safetensors
scipy aiohttp
tqdm yarl
psutil pyyaml
kornia pydantic
numba pydantic-settings
opencv4 pillow
GitPython scipy
numexpr tqdm
matplotlib psutil
pandas av
imageio-ffmpeg sqlalchemy
scikit-image soundfile
pip ]
accelerate ));
];
};
in rec { in rec {
packages = rec {
comfyui = pkgs.stdenv.mkDerivation {
name = "comfyui";
src = comfyuigit;
installPhase = ''
mkdir -p $out/bin
cp -r $src/* $out
echo "${pyenv}/bin/python main.py" > $out/bin/comfyui packages.comfyui_frontend = pkgs.python3Packages.buildPythonPackage rec {
chmod +x $out/bin/comfyui pname = "comfyui-frontend-package";
''; version = "1.23.4";
pyproject = true;
src = pkgs.fetchPypi {
pname = "comfyui_frontend_package";
inherit version;
sha256 = "sha256-8HrnNEO5rOuUez1MEzWUOJ512nraRTfH+BRJD5PwMfg=";
}; };
default = comfyui; buildInputs = [
python
python.pkgs.pip
];
}; };
devShell = pkgs.mkShell { packages.comfyui_workflow = pkgs.python3Packages.buildPythonPackage rec {
buildInputs = [packages.comfyuimain pyenv]; 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
'';
}; };
} }
); );
} }