packages: change obs-backgrounremoval override
This commit is contained in:
@@ -9,23 +9,28 @@ let
|
||||
|
||||
onnxruntime-gpu = (pkgs.onnxruntime.override { cudaSupport = withCuda; });
|
||||
|
||||
obs-backgrounremoval-gpu = (
|
||||
obs-backgroundremoval-gpu = (
|
||||
(pkgs.obs-studio-plugins.obs-backgroundremoval.override { onnxruntime = onnxruntime-gpu; })
|
||||
.overrideAttrs
|
||||
(old: {
|
||||
version = "1.1.14-beta";
|
||||
src = pkgs.fetchFromGitHub {
|
||||
owner = "occ-ai";
|
||||
repo = "obs-backgroundremoval";
|
||||
rev = "012a7f45fe4cb5363abee654d05c5cba4235feb5";
|
||||
hash = "sha256-ud9RfnbMXfOaIhkUYG7zyR4SxZhj3rZd9b4+8P4jBYs=";
|
||||
};
|
||||
cmakeFlags =
|
||||
if withCuda then
|
||||
(lib.lists.remove "-DDISABLE_ONNXRUNTIME_GPU=ON" old.cmakeFlags)
|
||||
else
|
||||
old.cmakeFlags;
|
||||
})
|
||||
(
|
||||
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
|
||||
{
|
||||
@@ -34,12 +39,10 @@ in
|
||||
package = pkgs.obs-studio.override {
|
||||
cudaSupport = withCuda;
|
||||
};
|
||||
plugins =
|
||||
with pkgs.obs-studio-plugins;
|
||||
[
|
||||
obs-pipewire-audio-capture
|
||||
obs-shaderfilter
|
||||
]
|
||||
++ lib.optionals withCuda [ obs-backgrounremoval-gpu ];
|
||||
plugins = with pkgs.obs-studio-plugins; [
|
||||
obs-pipewire-audio-capture
|
||||
obs-shaderfilter
|
||||
obs-backgroundremoval-gpu
|
||||
];
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user