add torch_ema to enviroment.yml

torchmd · May 13, 2024 · 7c11034 · 7c11034
1 parent 96934fe
commit 7c11034
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -17,3 +17,5 @@ dependencies:
   - pytest
   - psutil
   - gxx<12
+  - pip:
+    - torch-ema