release: 完成了mvp,具有基本的模型训练,语料生成和推理功能
This commit is contained in:
9
schema/dataset.py
Normal file
9
schema/dataset.py
Normal file
@@ -0,0 +1,9 @@
|
||||
from pydantic import BaseModel, RootModel
|
||||
from typing import List
|
||||
|
||||
class QAPair(BaseModel):
|
||||
question: str
|
||||
response: str
|
||||
|
||||
class QAArray(RootModel):
|
||||
root: List[QAPair]
|
Reference in New Issue
Block a user