feat: 重构模型库,更新数据集处理逻辑,优化导入路径,添加新的化学数据集类

This commit is contained in:
2025-01-07 13:43:56 +08:00
parent 5b09d27920
commit 0b71bfc617
8 changed files with 567 additions and 55 deletions
-1
View File
@@ -1,7 +1,6 @@
from dataclasses import dataclass, field
from typing import Optional
from trl import ScriptArguments, ModelConfig
from transformers import TrainingArguments
@dataclass