107 lines
3.3 KiB
Nix
107 lines
3.3 KiB
Nix
{
|
|
description = "Flake using pyproject.toml metadata";
|
|
|
|
inputs = {
|
|
nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
|
|
pyproject-nix = {
|
|
url = "github:pyproject-nix/pyproject.nix";
|
|
inputs.nixpkgs.follows = "nixpkgs";
|
|
};
|
|
flake-utils.url = "github:numtide/flake-utils";
|
|
};
|
|
|
|
outputs = {
|
|
self,
|
|
nixpkgs,
|
|
flake-utils,
|
|
...
|
|
}@inputs: let
|
|
system = "x86_64-linux";
|
|
pkgs = nixpkgs.legacyPackages.${system};
|
|
project = inputs.pyproject-nix.lib.project.loadPyproject {
|
|
projectRoot = ./.;
|
|
};
|
|
projectName = project.pyproject.project.name;
|
|
python = pkgs.python3;
|
|
|
|
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=";
|
|
};
|
|
|
|
phases = [ "installPhase" ];
|
|
|
|
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/models/"
|
|
unar -D -o $out/var/lib/insightface/models/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.mkPythonEditablePackage { inherit python; };
|
|
pythonEnv = python.pkgs.mkPythonEditablePackage ( arg );
|
|
in pkgs.mkShell {
|
|
packages = with pkgs; [
|
|
python3
|
|
uv
|
|
gcc14
|
|
python3Packages.fastapi
|
|
] ++ [protestswap];
|
|
shellHook = /*shell*/
|
|
''
|
|
ls -l ${buffaloModel}
|
|
|
|
INSIGHTFACE_ROOT_DIR=$(mktemp -d /tmp/insightface.XXXXXXXX)
|
|
export INSIGHTFACE_ROOT_DIR
|
|
|
|
mkdir -p "$INSIGHTFACE_ROOT_DIR/models/"
|
|
|
|
ln -s ${inswapperModel}/var/lib/insightface/models/inswapper_128.onnx "$INSIGHTFACE_ROOT_DIR/models/"
|
|
ln -s ${buffaloModel}/var/lib/insightface/models/buffalo_l "$INSIGHTFACE_ROOT_DIR/models/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}.default = protestswap;
|
|
};
|
|
}
|