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

调整mattergen-wrapper位置

Browse Source
This commit is contained in:
lzy
2025-04-02 16:57:37 +08:00
parent 7034566ee6
commit bac8f067e0
9 changed files with 24 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
mars_toolkit/services/__pycache__/mattergen_service.cpython-310.pyc
View File

Binary file not shown.

16
mars_toolkit/services/mattergen_service.py
View File

@@ -14,14 +14,14 @@ from typing import Dict, Any, Optional, Union, List
import threading
# 导入mattergen相关模块
import sys
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../../')))
from mattergen_wrapper import generator
CrystalGenerator = generator.CrystalGenerator
from mattergen.common.data.types import TargetProperty
from mattergen.common.utils.eval_utils import MatterGenCheckpointInfo
from mattergen.common.utils.data_classes import PRETRAINED_MODEL_NAME
# import sys
# sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../../')))
# from mars_toolkit.core.mattergen_wrapper import generator
# CrystalGenerator = generator.CrystalGenerator
# from mattergen.common.data.types import TargetProperty
# from mattergen.common.utils.eval_utils import MatterGenCheckpointInfo
# from mattergen.common.utils.data_classes import PRETRAINED_MODEL_NAME
from ..core.mattergen_wrapper import *
# 导入mars_toolkit配置
from mars_toolkit.core.config import config