Spaces:
Running
Running
File size: 779 Bytes
acd7cf4 4b2a9c2 acd7cf4 43d27f2 acd7cf4 4b2a9c2 acd7cf4 2a0edfe acd7cf4 2a0edfe acd7cf4 2a0edfe acd7cf4 2a0edfe acd7cf4 43d27f2 2a0edfe acd7cf4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from dataclasses import dataclass
from typing import Any
@dataclass
class WebuiParams:
"""
GraphGen parameters
"""
if_trainee_model: bool
upload_file: Any # gr.File
tokenizer: str
synthesizer_model: str
synthesizer_url: str
trainee_model: str
trainee_url: str
api_key: str
trainee_api_key: str
chunk_size: int
chunk_overlap: int
quiz_samples: int
partition_method: str
dfs_max_units: int
bfs_max_units: int
leiden_max_size: int
leiden_use_lcc: bool
leiden_random_seed: int
ece_max_units: int
ece_min_units: int
ece_max_tokens: int
ece_unit_sampling: str
mode: str
data_format: str
rpm: int
tpm: int
token_counter: Any
|