mirror of
https://github.com/denismhz/flake.git
synced 2025-11-09 16:16:23 +01:00
65 lines
1.6 KiB
Nix
65 lines
1.6 KiB
Nix
{
|
|
src,
|
|
buildPythonPackage,
|
|
safetensors,
|
|
psutil,
|
|
einops,
|
|
transformers,
|
|
scipy,
|
|
torchsde,
|
|
pillow,
|
|
torch,
|
|
torchvision,
|
|
accelerate,
|
|
}:
|
|
buildPythonPackage {
|
|
pname = "ComfyUI";
|
|
format = "other";
|
|
version = "latest";
|
|
inherit src;
|
|
propagatedBuildInputs = [
|
|
accelerate
|
|
torchvision
|
|
torch
|
|
safetensors
|
|
psutil
|
|
einops
|
|
transformers
|
|
scipy
|
|
pillow
|
|
torchsde
|
|
];
|
|
|
|
buildPhase = ''
|
|
runHook preBuild
|
|
|
|
mkdir -p dist
|
|
cp -R . $out
|
|
chmod -R +w $out
|
|
cd $out
|
|
substituteInPlace ./folder_paths.py --replace 'os.path.join(base_path, "models")' '"/home/denis/comfyui/models"'
|
|
substituteInPlace ./folder_paths.py --replace 'os.path.join(os.path.dirname(os.path.realpath(__file__)), "temp")' '"/home/denis/comfyui/temp"'
|
|
substituteInPlace ./folder_paths.py --replace 'os.path.join(os.path.dirname(os.path.realpath(__file__)), "output")' '"/home/denis/comfyui/output"'
|
|
substituteInPlace ./folder_paths.py --replace 'os.path.join(os.path.dirname(os.path.realpath(__file__)), "input")' '"/home/denis/comfyui/input"'
|
|
#make main.py executable > shebang
|
|
mkdir -p $out/bin
|
|
cat <<-EOF > main.py
|
|
$(echo "#!/usr/bin/python")
|
|
$(cat main.py)
|
|
EOF
|
|
chmod +x main.py
|
|
makeWrapper "$out/main.py" $out/bin/main-wrapped.py \
|
|
--set-default PYTHONPATH $PYTHONPATH \
|
|
|
|
rm -rf dist
|
|
|
|
runHook postBuild
|
|
'';
|
|
|
|
meta = {
|
|
description = "The most powerful and modular stable diffusion GUI and backend.";
|
|
homepage = "https://github.com/comfyanonymous/ComfyUI.git";
|
|
mainProgram = "main-wrapped.py";
|
|
};
|
|
}
|