pin libneuronxla

huggingface · Sep 18, 2024 · cb11276 · cb11276
1 parent af40e39
commit cb11276
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -70,6 +70,7 @@
         "torch==2.1.2.*",
         "torchvision==0.16.*",
         "neuronx_distributed==0.8.0",
+        "libneuronxla==2.0.2335",
     ],
     "diffusers": ["diffusers>=0.28.0, <0.29.0", "peft"],
     "sentence-transformers": ["sentence-transformers >= 2.2.0"],