diff --git a/requirements.txt b/requirements.txt index 967f3e3..81a4de8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,11 @@ -h5py -pytorch-lightning -matplotlib -numpy -PyYAML -scikit-learn -timm -torch -onnx -onnxruntime ---index-url https://git.riahub.ai/api/packages/qoherent/pypi/simple/ -utils \ No newline at end of file +h5py~=3.14 +pytorch-lightning~=2.5 +matplotlib~=3.8 +numpy~=1.26 +PyYAML~=6.0 +scikit-learn~=1.4 +timm~=0.9 +torch~=2.2 +onnx~=1.16 +onnxruntime~=1.18 +--index-url https://git.riahub.ai/api/packages/qoherent/pypi/simple \ No newline at end of file