修改mp的实现

2025-01-06 10:03:38 +08:00
parent 5380ee5f9e
commit c2417fec25
3 changed files with 72 additions and 8 deletions
--- a/constant.py
+++ b/constant.py
@@ -1,10 +1,14 @@
-TOPK_RESULT = 5
-TIME_OUT = 30
+TOPK_RESULT = 1
+TIME_OUT = 60

 # MP Configuration
 MP_API_KEY = "gfBp2in8qxm9Xm2SwLKFwNxDyZvNTAEt"
 MP_ENDPOINT = "https://api.materialsproject.org/"

+# Proxy
+HTTP_PROXY = "http://127.0.0.1:7897" 
+HTTPS_PROXY = "http://127.0.0.1:7897"
+
 # Model
 FAIRCHEM_MODEL_PATH = "/home/ubuntu/sas0/LYT/mars1215/mars_toolkit/model/eqV2_86M_omat_mp_salex.pt"
 FMAX = 0.05