updated requirements

Author: screemix

README.md

@@ -22,4 +22,4 @@ Download mappings and prepare relations' embeddings:
 
 Run verbalization:
 
-``` python3 verbalization.py --relation_vectors_path "data/embeddings/fasttext_vecs-wikidata5m.npy" --rel2ind_path "data/relation2ind-wikidata5m.json" --entity_mapping_path "data/mappings/wd5m_aliases_entities_v3.txt" --relation_mapping_path "data/relation2text-wikidata5m.json" --mongodb_port 27017```
+```python3 verbalization.py --relation_vectors_path "data/embeddings/fasttext_vecs-wikidata5m.npy" --rel2ind_path "data/relation2ind-wikidata5m.json" --entity_mapping_path "data/mappings/wd5m_aliases_entities_v3.txt" --relation_mapping_path "data/relation2text-wikidata5m.json" --mongodb_port 27017```

requirements.txt

@@ -1,4 +1,4 @@
-numpy==1.24.4
+numpy==1.21.1
 tiktoken==0.4.0
 tokenizers==0.13.3
 torch==2.1.0
@@ -37,4 +37,6 @@ sentencepiece==0.1.97
 sklearn==0.0.post1
 tensorboard==2.11.0
 sh==2.0.6
-wget==3.2
+wget==3.2
+python-dotenv==1.0.0
+horovod==0.28.1