mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-27 16:10:30 +08:00
Lewis Tunstall
23 lines
438 B
Python
23 lines
438 B
Python
from dataclasses import dataclass
|
|
|
|
import datasets
|
|
|
|
|
|
@dataclass
|
|
class OpenAssistantConfig(datasets.BuilderConfig):
|
|
"""BuilderConfig for OpenAssistant datasets."""
|
|
|
|
name: str = None
|
|
version: datasets.Version = None
|
|
description: str = None
|
|
schema: str = None
|
|
subset_id: str = None
|
|
|
|
|
|
lm_features = datasets.Features(
|
|
{
|
|
"text": datasets.Value("string"),
|
|
"meta": [datasets.Value("string")],
|
|
}
|
|
)
|