49 lines
1.2 KiB
Nix
49 lines
1.2 KiB
Nix
{
|
|
pkgs,
|
|
lib,
|
|
osConfig,
|
|
...
|
|
}:
|
|
let
|
|
withCuda = osConfig.system.name == "gunter";
|
|
|
|
onnxruntime-gpu = (pkgs.onnxruntime.override { cudaSupport = withCuda; });
|
|
|
|
obs-backgroundremoval-gpu = (
|
|
(pkgs.obs-studio-plugins.obs-backgroundremoval.override { onnxruntime = onnxruntime-gpu; })
|
|
.overrideAttrs
|
|
(
|
|
final: prev: {
|
|
version = "1.3.3";
|
|
src = pkgs.fetchFromGitHub {
|
|
owner = "royshil";
|
|
repo = "obs-backgroundremoval";
|
|
rev = final.version;
|
|
hash = "sha256-NDe71iDnVcnMilGr5kdbemq8jEKd3WW45tbMwxjqUwo=";
|
|
};
|
|
nativeBuildInputs = prev.nativeBuildInputs ++ [ pkgs.pkg-config ];
|
|
cmakeFlags = [
|
|
"--preset ubuntu-x86_64"
|
|
"-DCMAKE_MODULE_PATH:PATH=${final.src}/cmake"
|
|
"-DUSE_SYSTEM_ONNXRUNTIME=ON"
|
|
"-DVCPKG_TARGET_TRIPLET="
|
|
"-DUSE_PKGCONFIG=ON"
|
|
];
|
|
}
|
|
)
|
|
);
|
|
in
|
|
{
|
|
programs.obs-studio = {
|
|
enable = true;
|
|
package = pkgs.obs-studio.override {
|
|
cudaSupport = withCuda;
|
|
};
|
|
plugins = with pkgs.obs-studio-plugins; [
|
|
obs-pipewire-audio-capture
|
|
obs-shaderfilter
|
|
obs-backgroundremoval-gpu
|
|
];
|
|
};
|
|
}
|