nemo_automodel.components.launcher.skypilot.config

Module Contents

Classes

SkyPilotConfig

Data

SUPPORTED_CLOUDS

API

nemo_automodel.components.launcher.skypilot.config.SUPPORTED_CLOUDS

(‘aws’, ‘gcp’, ‘azure’, ‘lambda’, ‘kubernetes’)

class nemo_automodel.components.launcher.skypilot.config.SkyPilotConfig

cloud: str

‘field(…)’

accelerators: str

‘field(…)’

num_nodes: int

‘field(…)’

use_spot: bool

‘field(…)’

disk_size: int

‘field(…)’

instance_type: str | None

‘field(…)’

region: str | None

‘field(…)’

zone: str | None

‘field(…)’

job_name: str

‘field(…)’

setup: str

‘field(…)’

hf_home: str

‘field(…)’

hf_token: str

‘field(…)’

wandb_key: str

‘field(…)’

env_vars: dict[str, str]

‘field(…)’

command: str

‘field(…)’

__post_init__() → None