Improve obs slightly
Some checks are pending
pre-commit / pre-commit (push) Waiting to run

This commit is contained in:
Torjus Håkestad 2024-07-16 01:40:36 +02:00
parent 2bece06f10
commit 4c2d93eae7

View File

@ -1,14 +1,15 @@
{ pkgs, lib, osConfig, ... }: { pkgs, lib, osConfig, ... }:
let let
withCuda = if (osConfig.system.name == "gunter") then true else false; withCuda = osConfig.system.name == "gunter";
onnxruntime-gpu = (pkgs.onnxruntime.override { onnxruntime-gpu = (pkgs.onnxruntime.override {
cudaSupport = withCuda; cudaSupport = withCuda;
}).overrideAttrs (old: { }).overrideAttrs
# TODO: Remove when fixed in nixpkgs (old: {
# https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154 # TODO: Remove when fixed in nixpkgs
buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ]; # https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154
}); buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ];
});
obs-backgrounremoval-gpu = ((pkgs.obs-studio-plugins.obs-backgroundremoval.override { obs-backgrounremoval-gpu = ((pkgs.obs-studio-plugins.obs-backgroundremoval.override {
onnxruntime = onnxruntime-gpu; onnxruntime = onnxruntime-gpu;