@@ -27,6 +27,7 @@ class ClusterConfig:
27
27
k8s_pull_policy : str = "IfNotPresent"
28
28
k8s_job_ttl_seconds : int = 3600
29
29
k8s_backoff_limit : int = 3
30
+ k8s_remote : bool = False
30
31
31
32
# Cloud provider settings for remote Kubernetes
32
33
cloud_provider : str = "manual" # manual, aws, azure, gcp
@@ -38,18 +39,27 @@ class ClusterConfig:
38
39
aws_access_key_id : Optional [str ] = None
39
40
aws_secret_access_key : Optional [str ] = None
40
41
eks_cluster_name : Optional [str ] = None
42
+ aws_region : Optional [str ] = None
41
43
42
44
# Azure-specific settings
43
45
azure_subscription_id : Optional [str ] = None
44
46
azure_resource_group : Optional [str ] = None
45
47
azure_tenant_id : Optional [str ] = None
46
48
aks_cluster_name : Optional [str ] = None
49
+ azure_region : Optional [str ] = None
47
50
48
51
# GCP-specific settings
49
52
gcp_project_id : Optional [str ] = None
50
53
gcp_zone : Optional [str ] = None
51
54
gcp_service_account_key : Optional [str ] = None
52
55
gke_cluster_name : Optional [str ] = None
56
+ gcp_region : Optional [str ] = None
57
+
58
+ # Lambda Cloud settings
59
+ lambda_instance_type : Optional [str ] = None
60
+
61
+ # Hugging Face Spaces settings
62
+ hf_hardware : Optional [str ] = None
53
63
54
64
# Resource defaults
55
65
default_cores : int = 4
0 commit comments