diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..2268e08 --- /dev/null +++ b/environment.yml @@ -0,0 +1,19 @@ +name: ammico-dev +channels: + - pytorch + - nvidia + - rapidsai + - conda-forge + - defaults + +dependencies: + - python=3.11 + - cudatoolkit=11.8 + - pytorch=2.3.1 + - pytorch-cuda=11.8 + - torchvision=0.18.1 + - torchaudio=2.3.1 + - faiss-gpu-raft=1.8.0 + - ipykernel + - jupyterlab + - jupyterlab_widgets \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index 4cad313..1507f8b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,16 +18,20 @@ classifiers = [ "Operating System :: OS Independent", "License :: OSI Approved :: MIT License", ] - + dependencies = [ + "accelerate>=0.22", + "bitsandbytes", "colorgram.py", "colour-science", "dash", "dash-bootstrap-components", + "decord", "deepface", "google-cloud-vision", - "googletrans==4.0.0rc1", + "googletrans-py", # instead of googletrans4.0.0rc1, for a temporary solution due the incompatibility with jupyterlab "grpcio", + "huggingface-hub>=0.34.0", "importlib_metadata", "importlib_resources", "matplotlib", @@ -36,12 +40,15 @@ dependencies = [ "pandas", "Pillow", "pooch", + "qwen-vl-utils[decord]==0.0.8", "retina_face", + "safetensors>=0.6.2", "setuptools", "spacy", - "tensorflow<=2.16.0", + "tensorflow<2.15", # instead of <=2.16.0 to make it compatible with CUDA 11.8, may change after updating CUDA version. "tf-keras", "tqdm", + "transformers>=4.54", "webcolors", ]