add requirements

sangethmathewjohn · Jun 1, 2020 · b41948f · b41948f
1 parent fb8f427
commit b41948f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/movement-pruning/requirements.txt b/examples/movement-pruning/requirements.txt
@@ -0,0 +1,6 @@
+torch>=1.4.0
+-e git+https://github.com/huggingface/transformers.git@352d5472b0c1dec0f420d606d16747d851b4bda8#egg=transformers
+knockknock>=0.1.8.1
+h5py>=2.10.0
+numpy>=1.18.2
+scipy>=1.4.1