torjus/nixos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Add cuda for obs-backgroundremoval

Browse Source
This commit is contained in:
Torjus Håkestad 2024-05-28 19:44:52 +02:00
parent fcec0641aa
commit 9704525fdf
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
home/programs/obs-studio/default.nix
View File

@ -1,25 +1,31 @@
{ pkgs, ... }:
# let
# obs-backgroundremoval_1_1_10 = (pkgs.obs-studio-plugins.obs-backgroundremoval.overrideAttrs {
# version = "1.1.10";
# src = pkgs.fetchFromGitHub {
# owner = "occ-ai";
# repo = "obs-backgroundremoval";
# rev = "1.1.10";
# hash = "sha256-6OTtmjXS3fiorEPUSjjqrS8u9fVJnnZZoD722W3JQO8=";
# };
# }).override {
# onnxruntime = pkgs.unstable.onnxruntime;
# };
# in
{ pkgs, lib, osConfig, ... }:
let
withCuda = if (osConfig.system.name == "gunter") then true else false;
onnxruntime-gpu = (pkgs.onnxruntime.override {
cudaSupport = withCuda;
});
obs-backgrounremoval-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 = "94be8c35fe077be93a6f5ef347a802295a36dddd";
hash = "sha256-qnxDNeTWQYiRMqT6jNp8GC8ef6aaAAY+OXAak54dVc8=";
};
cmakeFlags =
if withCuda then
(lib.lists.remove "-DDISABLE_ONNXRUNTIME_GPU=ON" old.cmakeFlags)
else old.cmakeFlags;
}));
in
{
programs.obs-studio = {
enable = true;
plugins = with pkgs.obs-studio-plugins; [
obs-backgroundremoval
obs-pipewire-audio-capture
obs-shaderfilter
];
#plugins = [ obs-backgroundremoval_1_1_10 ];
] ++ [ obs-backgrounremoval-gpu ];
};
}