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