Преглед на файлове

genai plugin: add wheel package (#4626)

zhang-prog преди 1 месец
родител
ревизия
a6b11945fe
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      paddlex/paddlex_cli.py

+ 1 - 0
paddlex/paddlex_cli.py

@@ -369,6 +369,7 @@ def install(args):
         for plugin_type in plugin_types:
             if "vllm" in plugin_type or "sglang" in plugin_type:
                 try:
+                    install_packages(["wheel"], constraints="required")
                     if "vllm" in plugin_type:
                         install_packages(
                             ["flash-attn == 2.8.3"], constraints="required"