Compare commits
9 Commits
00ed153b61
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| 6e3f42ee71 | |||
| eb6191baf8 | |||
| 762895a2e0 | |||
| d9a945613c | |||
| 55d659e4e3 | |||
| 2b4621e32a | |||
| 5eb7e39562 | |||
| 94328d682d | |||
| b8f8e4d6a6 |
61
flake.lock
generated
Normal file
61
flake.lock
generated
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
{
|
||||||
|
"nodes": {
|
||||||
|
"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": 1772773019,
|
||||||
|
"narHash": "sha256-E1bxHxNKfDoQUuvriG71+f+s/NT0qWkImXsYZNFFfCs=",
|
||||||
|
"owner": "nixos",
|
||||||
|
"repo": "nixpkgs",
|
||||||
|
"rev": "aca4d95fce4914b3892661bcb80b8087293536c6",
|
||||||
|
"type": "github"
|
||||||
|
},
|
||||||
|
"original": {
|
||||||
|
"owner": "nixos",
|
||||||
|
"ref": "nixos-unstable",
|
||||||
|
"repo": "nixpkgs",
|
||||||
|
"type": "github"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"root": {
|
||||||
|
"inputs": {
|
||||||
|
"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
|
||||||
|
}
|
||||||
294
flake.nix
Normal file
294
flake.nix
Normal file
@@ -0,0 +1,294 @@
|
|||||||
|
{
|
||||||
|
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
|
||||||
|
'';
|
||||||
|
};
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user