Unverified Commit a772a777 authored by kirillrdy's avatar kirillrdy Committed by GitHub
Browse files

python3Packages.vllm: add optional-dependencies.audio (#465397)

parents 5f8b437d 46fb62a8
Loading
Loading
Loading
Loading
+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 = [
  ]