Upload 3 files

poetry.lock

@@ -341,6 +341,16 @@ files = [
     {file = "numpy-1.25.2.tar.gz", hash = "sha256:fd608e19c8d7c55021dffd43bfe5492fab8cc105cc8986f813f8c3c048b38760"},
 ]
 
+[[package]]
+name = "nvidia-nccl-cu12"
+version = "2.18.1"
+description = "NVIDIA Collective Communication Library (NCCL) Runtime"
+optional = false
+python-versions = ">=3"
+files = [
+    {file = "nvidia_nccl_cu12-2.18.1-py3-none-manylinux1_x86_64.whl", hash = "sha256:1a6c4acefcbebfa6de320f412bf7866de856e786e0462326ba1bac40de0b5e71"},
+]
+
 [[package]]
 name = "pillow"
 version = "10.1.0"
@@ -563,4 +573,4 @@ files = [
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "0e736806d7244b36780132f56a84854c8ae7c4b343ddbf118b5b4a3ab3da5390"
+content-hash = "c03e9996bdf1414d12b3683c65691b8a15dd2f54f04ff01ced175127452bd49d"

pyproject.toml

@@ -12,6 +12,7 @@ pillow = "^10.1.0"
 ftfy = "^6.1.1"
 torchvision = "^0.16.0"
 easydict = "^1.11"
+nvidia-nccl-cu12 = "2.18.1"
 
 
 [build-system]