diff --git a/olive/olive_config.json b/olive/olive_config.json index caee1298e..e3a728a42 100644 --- a/olive/olive_config.json +++ b/olive/olive_config.json @@ -394,7 +394,7 @@ "flash-attn": [ "flash_attn" ], "gpu": [ "onnxruntime-gpu" ], "inc": [ "neural-compressor" ], - "lora": [ "accelerate>=0.30.0", "peft>=0.12.0", "scipy" ], + "lora": [ "accelerate>=0.30.0", "peft", "scipy" ], "nvmo": [ "nvidia-modelopt", "onnx-graphsurgeon", "datasets>=2.14.4", "cppimport==22.8.2" ], "openvino": [ "openvino==2023.2.0", "nncf==2.7.0", "numpy<2.0" ], "optimum": [ "optimum" ],