Commit 46fb62a8 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.vllm: add optional-dependencies.audio

parent a866f627

pkgs/development/python-modules/vllm/default.nix

+14 −0
Original line number Diff line number Diff line
@@ -75,6 +75,11 @@ 
  setproctitle,

  openai-harmony,



  # optional-dependencies

  # audio

  librosa,

  soundfile,



  # internal dependency - for overriding in overlays

  vllm-flash-attn ? null,
@@ -463,6 +468,15 @@ buildPythonPackage rec { 
    flashinfer

  ];



  optional-dependencies = {

    audio = [

      librosa

      soundfile

      mistral-common

    ]

    ++ mistral-common.optional-dependencies.audio;

  };



  dontUseCmakeConfigure = true;

  cmakeFlags = [

  ]