lzy/mars-mcp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mcp,生成数据代码

Browse Source
This commit is contained in:
lzy
2025-04-16 11:15:01 +08:00
parent 72045e5cfe
commit 6b92e54a41
66 changed files with 1938 additions and 1483 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mars_toolkit/core/config.py Normal file → Executable file
View File

@@ -22,12 +22,12 @@ class Config:
HTTPS_PROXY = 'http://192.168.168.1:20171'
# FairChem
FAIRCHEM_MODEL_PATH = '/home/ubuntu/50T/lzy/mars-mcp/pretrained_models/fairchem_ckpt/eqV2_86M_omat_mp_salex.pt'
FAIRCHEM_MODEL_PATH = '/home/ubuntu/50T/nfs/lzy/mars-mcp/pretrained_models/fairchem_ckpt/eqV2_86M_omat_mp_salex.pt'
FMAX = 0.05
# MatterGen
MATTERGENMODEL_ROOT = '/home/ubuntu/50T/lzy/mars-mcp/pretrained_models/mattergen_ckpt'
MATTERGEN_ROOT='/home/ubuntu/50T/lzy/mars-mcp/mattergen'
MATTERGENMODEL_ROOT = '/home/ubuntu/50T/nfs/lzy/mars-mcp/pretrained_models/mattergen_ckpt'
MATTERGEN_ROOT='/home/ubuntu/50T/nfs/lzy/mars-mcp/mattergen'
MATTERGENMODEL_RESULT_PATH = 'results/'
# Dify
@@ -38,7 +38,7 @@ class Config:
SEARXNG_HOST="http://192.168.168.1:40032/"
# Visualization
VIZ_CIF_OUTPUT_ROOT = '/home/ubuntu/50T/lzy/mars-mcp/outputs/cif_visualization'
VIZ_CIF_OUTPUT_ROOT = '/home/ubuntu/50T/nfs/lzy/mars-mcp/outputs/cif_visualization'
@classmethod
def as_dict(cls) -> Dict[str, Any]: