Spaces:
Sleeping
Sleeping
File size: 1,738 Bytes
33d4721 |
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 |
from typing import Dict, Optional
from pydantic import Field
from autotrain.trainers.common import AutoTrainParams
class GenericParams(AutoTrainParams):
"""
GenericParams is a class that holds configuration parameters for an AutoTrain SpaceRunner project.
Attributes:
username (str): The username for your Hugging Face account.
project_name (str): The name of the project.
data_path (str): The file path to the dataset.
token (str): The authentication token for accessing Hugging Face Hub.
script_path (str): The file path to the script to be executed. Path to script.py.
env (Optional[Dict[str, str]]): A dictionary of environment variables to be set.
args (Optional[Dict[str, str]]): A dictionary of arguments to be passed to the script.
"""
username: str = Field(
None, title="Hugging Face Username", description="The username for your Hugging Face account."
)
project_name: str = Field("project-name", title="Project Name", description="The name of the project.")
data_path: str = Field(None, title="Data Path", description="The file path to the dataset.")
token: str = Field(None, title="Hub Token", description="The authentication token for accessing Hugging Face Hub.")
script_path: str = Field(
None, title="Script Path", description="The file path to the script to be executed. Path to script.py"
)
env: Optional[Dict[str, str]] = Field(
None, title="Environment Variables", description="A dictionary of environment variables to be set."
)
args: Optional[Dict[str, str]] = Field(
None, title="Arguments", description="A dictionary of arguments to be passed to the script."
)
|