prunebutt/protestswap
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

finally works...?

Browse source
This commit is contained in:
Prunebutt 2025-10-19 22:31:59 +02:00
parent e047bcd136
commit 30ee8107fd
5 changed files with 115 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

92
flake.nix
View file

@ -24,52 +24,82 @@
projectName = project.pyproject.project.name;
python = pkgs.python3;
inswapperModel = pkgs.stdenv.mkDerivation {
name = "inswapper";
inswapperModel = pkgs.stdenv.mkDerivation {
name = "inswapper_128";
src = pkgs.fetchurl {
url = "https://huggingface.co/thebiglaskowski/inswapper_128.onnx/resolve/main/inswapper_128.onnx";
hash = "sha256-5KPwjHU8ty0E4Qqg99vj3uu/OVZ9Tq1tzgjpiqSeFq8=";
};
unpackPhase = ''true'';
phases = [
# "installPhase"
];
phases = [ "installPhase" ];
installPhase = let dest = "$out/var/lib/${projectName}"; in /*shell*/''
ls -l "$src"
mkdir -p "${dest}"
cp $src ${dest}/inswapper_128.onnx
installPhase = ''
mkdir -p "$out/var/lib/insightface/models/"
cp $src "$out/var/lib/insightface/models/inswapper_128.onnx"
'';
};
buffaloModel = pkgs.stdenv.mkDerivation {
name = "buffalo_l_Model";
src = pkgs.fetchurl {
url = "https://github.com/deepinsight/insightface/releases/download/v0.7/buffalo_l.zip";
hash = "sha256-gP/jfYpZQNWac4TCAaKjjUdB8vPFHu9G67KCGKewyi8=";
};
unpackPhase = /*shell*/''
ls -l $src
echo $src
mkdir -p "$out/var/lib/insightface/buffalo_l/"
unar -D -o $out/var/lib/insightface/buffalo_l/ $src
'';
nativeBuildInputs = [ pkgs.unar ];
};
postInstall = /*shell*/''
mkdir -p $out/var/lib/insightface/models/buffalo_l
cp ${inswapperModel} $out/var/lib/insightface/models/inswapper_128.onnx
'';
# ${pkgs.unar}/bin/unar -D -o $out/var/lib/insightface/models/buffalo_l ${buffaloModel}
# '';
packageAttrs = project.renderers.buildPythonPackage {
inherit python;
};
packageAttrsLive = project.renderers.mkPythonEditablePackage {
inherit python;
};
protestswap = python.pkgs.buildPythonPackage ( packageAttrs # // { inherit postInstall; }
);
in {
devShells.${system}.default = let
arg = project.renderers.withPackages { inherit python;};
pythonEnv = python.withPackages arg;
arg = project.renderers.mkPythonEditablePackage { inherit python; };
pythonEnv = python.pkgs.mkPythonEditablePackage ( arg );
in pkgs.mkShell {
packages = [ pythonEnv inswapperModel];
shellHook = ''
export FOO=${inswapperModel}/var/lib/${projectName}/inswapper_128.onnx
echo "Binary at '$FOO'"
packages = [ pkgs.python3 pkgs.uv
# buffaloModel
protestswap pkgs.gcc14 ];
shellHook = /*shell*/
''
ls -l ${buffaloModel}
INSIGHTFACE_ROOT_DIR=$(mktemp -d /tmp/insightface.XXXXXXXX)
export INSIGHTFACE_ROOT_DIR
mkdir -p "$INSIGHTFACE_ROOT_DIR"
ln -s ${inswapperModel}/var/lib/insightface/models "$INSIGHTFACE_ROOT_DIR/models"
ln -s ${buffaloModel}/var/lib/insightface/buffalo_l "$INSIGHTFACE_ROOT_DIR/buffalo_l"
# export PROTESTSWAP_ROOT="/tmp/protestswap_models"
# mkdir -p "$PROTESTSWAP_ROOT/buffalo_l"
#
# cp ${inswapperModel} "$PROTESTSWAP_ROOT/inswapper_128.onnx"
unset PYTHONPATH
uv sync
. .venv/bin/activate
'';
};
packages.${system} = let
attrs = project.renderers.buildPythonPackage {
inherit python;
};
protestswap = (python.pkgs.buildPythonPackage ( attrs // {
buildInputs = [ inswapperModel ];
} ));
in {
# inherit inswapperModel;
# "${projectName}" = protestswap;
default = pkgs.symlinkJoin {
name = "${projectName}";
paths = [inswapperModel protestswap];
};
};
packages.${system}.default = protestswap;
};
}