295 lines
9.6 KiB
Nix
295 lines
9.6 KiB
Nix
{
|
|
description = "ComfyUI as a Nix Flake";
|
|
|
|
inputs = {
|
|
nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
|
|
flake-utils.url = "github:numtide/flake-utils";
|
|
};
|
|
|
|
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
|
|
kornia
|
|
]
|
|
));
|
|
in rec {
|
|
|
|
packages.comfyui_frontend = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-frontend-package";
|
|
version = "1.39.19";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_frontend_package";
|
|
inherit version;
|
|
sha256 = "sha256-BLpdAQDwJlZlZ495lQOvRDs+LsV+W6/uj9wFzYhGLkU=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
|
|
patchPhase = ''
|
|
sed -i 's/or "0.1.0"/or "${version}"/' setup.py
|
|
'';
|
|
};
|
|
|
|
packages.comfyui_workflow_core = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-workflow-templates-core";
|
|
version = "0.3.159";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_workflow_templates_core";
|
|
inherit version;
|
|
sha256 = "sha256-GMsbXUsCaMMZL5voHCway1t5bf4oiTOUEOPoX0iemZY=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
};
|
|
|
|
packages.comfyui_workflow_media_api = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-workflow-templates-media-api";
|
|
version = "0.3.59";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_workflow_templates_media_api";
|
|
inherit version;
|
|
sha256 = "sha256-0CykAChG27GkZj5JRGqFhW7twVNyG//mPXVcVvxo3sw=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
};
|
|
|
|
packages.comfyui_workflow_media_video = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-workflow-templates-media-video";
|
|
version = "0.3.57";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_workflow_templates_media_video";
|
|
inherit version;
|
|
sha256 = "sha256-qs1tHUV8RelIHPkWtN1iCDlEl3tg47iSYlLQfeE6Vj4=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
};
|
|
|
|
packages.comfyui_workflow_media_image = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-workflow-templates-media-image";
|
|
version = "0.3.98";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_workflow_templates_media_image";
|
|
inherit version;
|
|
sha256 = "sha256-URjIPhOjMdrcpo4v9YHeTQ0KyCzbLgoOLMEO6ACeueQ=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
};
|
|
|
|
packages.comfyui_workflow_media_other = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-workflow-templates-media-other";
|
|
version = "0.3.131";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_workflow_templates_media_other";
|
|
inherit version;
|
|
sha256 = "sha256-cpFtJf4TiIAzat2XgMKQx7P2mlmdp+hcxnyIBVUHi4g=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
};
|
|
|
|
packages.comfyui_workflow = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-workflow-templates";
|
|
version = "0.9.11";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_workflow_templates";
|
|
inherit version;
|
|
sha256 = "sha256-bnO21Hbd/QnxtOYXcuH3GJEBivrd/m1X3m7Oz1XycCc=";
|
|
};
|
|
|
|
build-system = [ python.pkgs.setuptools ];
|
|
|
|
dependencies = [
|
|
packages.comfyui_workflow_core
|
|
packages.comfyui_workflow_media_api
|
|
packages.comfyui_workflow_media_video
|
|
packages.comfyui_workflow_media_image
|
|
packages.comfyui_workflow_media_other
|
|
];
|
|
|
|
# Create an empty templates directory to satisfy ComfyUI's expectation
|
|
# The newer workflow templates package no longer ships templates in this location
|
|
postInstall = ''
|
|
mkdir -p $out/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates/templates
|
|
cp -r ${packages.comfyui_workflow_media_video}/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates_media_video/templates $out/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates/templates/video
|
|
cp -r ${packages.comfyui_workflow_media_other}/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates_media_other/templates $out/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates/templates/other
|
|
cp -r ${packages.comfyui_workflow_media_image}/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates_media_image/templates $out/lib/python${python.pythonVersion}/site-packages/comfyui_workflow_templates/templates/image
|
|
'';
|
|
};
|
|
|
|
packages.comfyui_embedded_docs = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfyui-embedded-docs";
|
|
version = "0.4.3";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfyui_embedded_docs";
|
|
inherit version;
|
|
sha256 = "sha256-HMKr+T65Bv50bb8caScbH1ACFrCgqg1IoIQhjOtxu6w=";
|
|
};
|
|
|
|
build-system = [python.pkgs.setuptools ];
|
|
};
|
|
|
|
packages.spandrel = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "spandrel";
|
|
version = "0.4.1";
|
|
pyproject = true;
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "spandrel";
|
|
inherit version;
|
|
sha256 = "sha256-ZG2YFqlC5Z1WqrLckENTlS5X3uSyyz9Z9+pNwPsRofI=";
|
|
};
|
|
|
|
build-system = [
|
|
python.pkgs.setuptools
|
|
python.pkgs.torch
|
|
python.pkgs.torchvision
|
|
python.pkgs.safetensors
|
|
python.pkgs.numpy
|
|
python.pkgs.einops
|
|
python.pkgs.typing-extensions
|
|
];
|
|
};
|
|
|
|
packages.comfy_kitchen = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfy-kitchen";
|
|
version = "0.2.7";
|
|
format = "wheel";
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfy_kitchen";
|
|
inherit version;
|
|
format = "wheel";
|
|
dist = "py3";
|
|
python = "py3";
|
|
sha256 = "sha256-+PqlebadMx0vHqwJ6WqVWGwqa5WKVLwZ5/HBp3hS3TY=";
|
|
};
|
|
};
|
|
|
|
packages.comfy_aimdo = pkgs.python3Packages.buildPythonPackage rec {
|
|
pname = "comfy-aimdo";
|
|
version = "0.2.9";
|
|
format = "wheel";
|
|
|
|
src = pkgs.fetchPypi {
|
|
pname = "comfy_aimdo";
|
|
inherit version;
|
|
format = "wheel";
|
|
dist = "py3";
|
|
python = "py3";
|
|
sha256 = "sha256-nJpzj8zxaRXewIh6yHd/Um526o04SD0Pzg/N327IIag=";
|
|
};
|
|
};
|
|
|
|
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
|
|
ps.kornia
|
|
ps.requests
|
|
packages.comfyui_workflow_core
|
|
packages.comfyui_workflow_media_api
|
|
packages.comfyui_workflow_media_video
|
|
packages.comfyui_workflow_media_image
|
|
packages.comfyui_workflow_media_other
|
|
packages.comfyui_frontend
|
|
packages.comfyui_workflow
|
|
packages.comfyui_embedded_docs
|
|
packages.spandrel
|
|
packages.comfy_kitchen
|
|
packages.comfy_aimdo
|
|
]));
|
|
in pkgs.stdenv.mkDerivation {
|
|
pname = "ComfyUI";
|
|
version = "0.16.4";
|
|
|
|
src = pkgs.fetchFromGitHub {
|
|
owner = "Comfy-Org";
|
|
repo = "ComfyUI";
|
|
tag = "v0.16.4";
|
|
sha256 = "sha256-wPPsXQytzpb7gSzE4dUT5JME/q03Z2g6pk9RidHNy8A=";
|
|
};
|
|
|
|
buildInputs = [
|
|
python_extended
|
|
];
|
|
|
|
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
|
|
'';
|
|
};
|
|
}
|
|
);
|
|
}
|