cloud: update configuration wizard to show correct hardware type selector

simpletuner/simpletuner_sdk/server/services/cloud/pricing.py

@@ -245,12 +245,12 @@ def get_default_hardware(self) -> HardwareOption:
 REPLICATE_DEFAULT_HARDWARE = {
     "gpu-l40s": {
         "name": "L40S (48GB)",
-        "cost_per_second": 0.000975,
+        "cost_per_second": 0.000972222,
         "memory_gb": 48,
     },
-    "gpu-a100-large": {
-        "name": "A100 (80GB)",
-        "cost_per_second": 0.001400,
+    "gpu-h100": {
+        "name": "H100 (80GB)",
+        "cost_per_second": 0.001525,
         "memory_gb": 80,
     },
 }

simpletuner/simpletuner_sdk/server/services/cloud/provider_registry.py

@@ -93,7 +93,12 @@ async def get_enriched_providers() -> List[Dict[str, Any]]:
     - Hardware info and costs
     """
     from ...routes.cloud._shared import get_job_store
-    from .replicate_client import DEFAULT_MODEL, get_default_hardware_cost_per_hour, get_hardware_info_async
+    from .replicate_client import (
+        DEFAULT_HARDWARE_INFO,
+        DEFAULT_MODEL,
+        get_default_hardware_cost_per_hour,
+        get_hardware_info_async,
+    )
     from .replicate_profiles import (
         DEFAULT_REPLICATE_HARDWARE_PROFILE,
         get_replicate_hardware_profile,
@@ -120,6 +125,25 @@ async def get_enriched_providers() -> List[Dict[str, Any]]:
             hardware_info = await get_hardware_info_async(store)
             l40s_info = hardware_info.get(definition.default_hardware_id, {})
             cost_per_hour = await get_default_hardware_cost_per_hour(store)
+            profile_options = list_replicate_hardware_profiles()
+            base_costs = {
+                "h100": (hardware_info.get("gpu-h100") or DEFAULT_HARDWARE_INFO["gpu-h100"]).get("cost_per_second", 0.001525)
+                * 3600,
+                "l40s": (hardware_info.get("gpu-l40s") or DEFAULT_HARDWARE_INFO["gpu-l40s"]).get(
+                    "cost_per_second", 0.000972222
+                )
+                * 3600,
+            }
+            for option in profile_options:
+                base_profile = "h100" if option["id"].startswith("h100") else "l40s"
+                multiplier = 1
+                if "-x" in option["id"]:
+                    try:
+                        multiplier = int(option["id"].rsplit("-x", 1)[1])
+                    except (TypeError, ValueError):
+                        multiplier = 1
+                option["cost_per_hour"] = round(base_costs[base_profile] * multiplier, 2)
+                option["cost_per_second"] = (base_costs[base_profile] / 3600) * multiplier
 
             provider_data.update(
                 {
@@ -129,7 +153,7 @@ async def get_enriched_providers() -> List[Dict[str, Any]]:
                     "cost_per_hour": round(cost_per_hour, 2),
                     "configured": bool(get_secrets_manager().get_replicate_token()),
                     "hardware_profile": default_profile.id,
-                    "hardware_profiles": list_replicate_hardware_profiles(),
+                    "hardware_profiles": profile_options,
                 }
             )
 

simpletuner/simpletuner_sdk/server/services/cloud/replicate_client.py

@@ -41,10 +41,10 @@ def _get_credential_resolver():
 DEFAULT_MODEL = get_replicate_hardware_profile(DEFAULT_REPLICATE_HARDWARE_PROFILE).model
 
 # Default hardware info for cost estimation (fallback values if not configured)
-# These are based on Replicate's published pricing as of 2024
+# These are based on Replicate's published pricing.
 DEFAULT_HARDWARE_INFO: Dict[str, Dict[str, Any]] = {
-    "gpu-l40s": {"name": "L40S (48GB)", "cost_per_second": 0.000975},
-    "gpu-a100-large": {"name": "A100 (80GB)", "cost_per_second": 0.001400},
+    "gpu-l40s": {"name": "L40S (48GB)", "cost_per_second": 0.000972222},
+    "gpu-h100": {"name": "H100 (80GB)", "cost_per_second": 0.001525},
 }
 
 # Cached hardware info (loaded from config or defaults)
@@ -112,7 +112,7 @@ async def get_default_hardware_cost_per_hour(store: Optional[Any] = None) -> flo
     """
     hardware = await get_hardware_info_async(store)
     l40s = hardware.get("gpu-l40s", DEFAULT_HARDWARE_INFO["gpu-l40s"])
-    return l40s.get("cost_per_second", 0.000975) * 3600
+    return l40s.get("cost_per_second", 0.000972222) * 3600
 
 
 def update_hardware_info_cache(hardware_info: Dict[str, Dict[str, Any]]) -> None:

simpletuner/simpletuner_sdk/server/services/cloud/replicate_profiles.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
-from typing import Dict, List
+from typing import Any, Dict, List
 
 DEFAULT_REPLICATE_HARDWARE_PROFILE = "h100"
 
@@ -70,7 +70,7 @@ class ReplicateHardwareProfile:
 }
 
 
-def list_replicate_hardware_profiles() -> List[Dict[str, str]]:
+def list_replicate_hardware_profiles() -> List[Dict[str, Any]]:
     """Return profile metadata suitable for API/UI responses."""
     return [
         {

simpletuner/static/js/modules/cloud/job-submission.js

@@ -255,6 +255,66 @@ window.cloudSubmissionMethods = {
         ];
     },
 
+    getReplicateBaseHardwareOptions() {
+        return this.getReplicateHardwareProfiles()
+            .filter((profile) => profile.id === 'l40s' || profile.id === 'h100')
+            .sort((a, b) => (a.id === 'l40s' ? -1 : 1));
+    },
+
+    getSelectedReplicateBaseHardware() {
+        const selected = this.preSubmitModal?.hardwareProfile || this.getDefaultReplicateHardwareProfile();
+        return String(selected || '').startsWith('l40s') ? 'l40s' : 'h100';
+    },
+
+    getSelectedReplicateHardwareProfile() {
+        const selected = this.preSubmitModal?.hardwareProfile || this.getDefaultReplicateHardwareProfile();
+        const profiles = this.getReplicateHardwareProfiles();
+        return profiles.find((profile) => profile.id === selected) ||
+               profiles.find((profile) => profile.id === this.getSelectedReplicateBaseHardware()) ||
+               null;
+    },
+
+    setReplicateBaseHardwareProfile(profileId) {
+        if (!profileId) {
+            return;
+        }
+        const currentProfile = this.preSubmitModal?.hardwareProfile || this.getDefaultReplicateHardwareProfile();
+        const multiplier = String(currentProfile || '').match(/-x\d+$/)?.[0] || '';
+        const candidateProfile = `${profileId}${multiplier}`;
+        const profiles = this.getReplicateHardwareProfiles();
+        const nextProfile = profiles.some((profile) => profile.id === candidateProfile)
+            ? candidateProfile
+            : profileId;
+        this.preSubmitModal.hardwareProfile = nextProfile;
+        this.saveHardwareProfile(nextProfile);
+    },
+
+    getReplicateBaseHardwareCostDisplay() {
+        const selected = this.getSelectedReplicateBaseHardware();
+        const profile = this.getSelectedReplicateHardwareProfile() ||
+                        this.getReplicateBaseHardwareOptions().find((option) => option.id === selected);
+        if (typeof profile?.cost_per_hour === 'number') {
+            return '$' + profile.cost_per_hour.toFixed(2) + '/hr';
+        }
+        if (selected === 'h100') {
+            return '$5.49/hr';
+        }
+        return '$3.50/hr';
+    },
+
+    getReplicateBaseHardwareCostDetail() {
+        const selected = this.getSelectedReplicateBaseHardware();
+        const profile = this.getSelectedReplicateHardwareProfile() ||
+                        this.getReplicateBaseHardwareOptions().find((option) => option.id === selected);
+        if (typeof profile?.cost_per_second === 'number') {
+            return '$' + profile.cost_per_second.toFixed(6) + '/sec';
+        }
+        if (selected === 'h100') {
+            return '$0.001525/sec';
+        }
+        return '$0.000972/sec';
+    },
+
     getDefaultReplicateHardwareProfile() {
         const storedProfile = localStorage.getItem('cloud_replicate_hardware_profile');
         if (storedProfile) {

simpletuner/templates/cloud_tab.html

@@ -161,18 +161,47 @@ <h6 class="mb-0"><i class="fas fa-download me-2"></i>Local Outputs</h6>
                     <h6 class="mb-0"><i class="fas fa-server me-2"></i>Hardware</h6>
                 </div>
                 <div class="cloud-card-body py-2">
-                    <template x-for="provider in providers.filter(p => p.id === activeProvider)" :key="provider.id">
+                    <template x-if="activeProvider === 'replicate'">
                         <div class="small">
-                            <div class="d-flex justify-content-between mb-1">
-                                <span class="text-muted">GPU</span>
-                                <span class="text-info" x-text="provider.hardware || 'N/A'"></span>
+                            <div class="mb-2">
+                                <span class="text-muted d-block mb-1">Hardware</span>
+                                <div class="d-flex gap-2">
+                                    <template x-for="profile in getReplicateBaseHardwareOptions()" :key="profile.id">
+                                        <button type="button"
+                                                class="btn btn-sm rounded-pill flex-fill"
+                                                data-testid="cloud-settings-hardware-profile"
+                                                :class="getSelectedReplicateBaseHardware() === profile.id ? 'btn-info text-dark' : 'btn-outline-secondary'"
+                                                :aria-pressed="getSelectedReplicateBaseHardware() === profile.id ? 'true' : 'false'"
+                                                @click="setReplicateBaseHardwareProfile(profile.id)">
+                                            <span x-text="profile.label || profile.id.toUpperCase()"></span>
+                                        </button>
+                                    </template>
+                                </div>
                             </div>
-                            <div class="d-flex justify-content-between">
-                                <span class="text-muted">Cost</span>
-                                <span x-text="provider.cost_per_hour ? ('$' + provider.cost_per_hour.toFixed(2) + '/hr') : 'N/A'"></span>
+                            <div class="d-flex justify-content-between align-items-start">
+                                <span class="text-muted">Cost (per hour)</span>
+                                <span class="text-end">
+                                    <span x-text="getReplicateBaseHardwareCostDisplay()"></span>
+                                    <span class="d-block text-muted" style="font-size: 0.7rem;"
+                                          x-text="getReplicateBaseHardwareCostDetail()"></span>
+                                </span>
                             </div>
                         </div>
                     </template>
+                    <div x-show="activeProvider !== 'replicate'">
+                        <template x-for="provider in providers.filter(p => p.id === activeProvider)" :key="provider.id">
+                            <div class="small">
+                                <div class="d-flex justify-content-between mb-1">
+                                    <span class="text-muted">GPU</span>
+                                    <span class="text-info" x-text="provider.hardware || 'N/A'"></span>
+                                </div>
+                                <div class="d-flex justify-content-between">
+                                    <span class="text-muted">Cost</span>
+                                    <span x-text="provider.cost_per_hour ? ('$' + provider.cost_per_hour.toFixed(2) + '/hr') : 'N/A'"></span>
+                                </div>
+                            </div>
+                        </template>
+                    </div>
                 </div>
             </div>
 

simpletuner/templates/partials/cloud_hero_cta.html

@@ -21,7 +21,7 @@ <h2 class="hero-title">Get Started with Replicate</h2>
                 <div class="hero-features">
                     <div class="hero-feature">
                         <i class="fas fa-bolt text-warning"></i>
-                        <span>L40S & A100 GPUs</span>
+                        <span>L40S & H100 GPUs</span>
                     </div>
                     <div class="hero-feature">
                         <i class="fas fa-dollar-sign text-success"></i>

simpletuner/templates/partials/cloud_onboarding_flow.html

@@ -224,11 +224,28 @@ <h3>What This Will Cost</h3>
                     <div class="cost-breakdown">
                         <div class="cost-row">
                             <span class="cost-label">Hardware</span>
-                            <span class="cost-value">L40S (48GB VRAM)</span>
+                            <span class="cost-value">
+                                <span class="d-flex gap-2 justify-content-end">
+                                    <template x-for="profile in getReplicateBaseHardwareOptions()" :key="profile.id">
+                                        <button type="button"
+                                                class="btn btn-sm rounded-pill"
+                                                data-testid="cloud-onboarding-hardware-profile"
+                                                :class="getSelectedReplicateBaseHardware() === profile.id ? 'btn-info text-dark' : 'btn-outline-secondary'"
+                                                :aria-pressed="getSelectedReplicateBaseHardware() === profile.id ? 'true' : 'false'"
+                                                @click="setReplicateBaseHardwareProfile(profile.id)">
+                                            <span x-text="profile.label || profile.id.toUpperCase()"></span>
+                                        </button>
+                                    </template>
+                                </span>
+                            </span>
                         </div>
                         <div class="cost-row">
-                            <span class="cost-label">Rate</span>
-                            <span class="cost-value">~$3.50/hour</span>
+                            <span class="cost-label">Cost (per hour)</span>
+                            <span class="cost-value">
+                                <span x-text="getReplicateBaseHardwareCostDisplay()"></span>
+                                <span class="d-block text-muted" style="font-size: 0.7rem;"
+                                      x-text="getReplicateBaseHardwareCostDetail()"></span>
+                            </span>
                         </div>
                         <div class="cost-row">
                             <span class="cost-label">Typical LoRA (2000 steps)</span>

tests/js/job_submission.test.js

@@ -421,6 +421,76 @@ describe('cloudSubmissionMethods', () => {
                 'l40s-x8',
             ]);
         });
+
+        test('base hardware options expose l40s and h100 buttons', () => {
+            context.providers = [{
+                id: 'replicate',
+                hardware_profiles: [
+                    { id: 'h100', label: 'H100', cost_per_hour: 5.49, cost_per_second: 0.001525 },
+                    { id: 'h100-x2', label: '2x H100', cost_per_hour: 10.98, cost_per_second: 0.00305 },
+                    { id: 'l40s', label: 'L40S', cost_per_hour: 3.50, cost_per_second: 0.000972222 },
+                    { id: 'l40s-x2', label: '2x L40S', cost_per_hour: 7.00, cost_per_second: 0.001944444 },
+                ],
+            }];
+
+            const options = context.getReplicateBaseHardwareOptions();
+
+            expect(options.map((option) => option.id)).toEqual(['l40s', 'h100']);
+        });
+
+        test('settings hardware selector preserves existing gpu count multiplier', () => {
+            context.providers = [{
+                id: 'replicate',
+                hardware_profiles: [
+                    { id: 'h100', label: 'H100' },
+                    { id: 'h100-x4', label: '4x H100' },
+                    { id: 'l40s', label: 'L40S' },
+                    { id: 'l40s-x4', label: '4x L40S' },
+                ],
+            }];
+            context.preSubmitModal.hardwareProfile = 'h100-x4';
+
+            context.setReplicateBaseHardwareProfile('l40s');
+
+            expect(context.preSubmitModal.hardwareProfile).toBe('l40s-x4');
+            expect(localStorage.getItem('cloud_replicate_hardware_profile')).toBe('l40s-x4');
+        });
+
+        test('settings hardware selector falls back to base profile when matching multiplier is unavailable', () => {
+            context.providers = [{
+                id: 'replicate',
+                hardware_profiles: [
+                    { id: 'h100', label: 'H100' },
+                    { id: 'h100-x4', label: '4x H100' },
+                    { id: 'l40s', label: 'L40S' },
+                ],
+            }];
+            context.preSubmitModal.hardwareProfile = 'h100-x4';
+
+            context.setReplicateBaseHardwareProfile('l40s');
+
+            expect(context.preSubmitModal.hardwareProfile).toBe('l40s');
+            expect(localStorage.getItem('cloud_replicate_hardware_profile')).toBe('l40s');
+        });
+
+        test('hardware cost display uses provider supplied pricing', () => {
+            context.providers = [{
+                id: 'replicate',
+                hardware_profiles: [
+                    { id: 'h100', label: 'H100', cost_per_hour: 6.66, cost_per_second: 0.001851852 },
+                    { id: 'h100-x4', label: '4x H100', cost_per_hour: 26.64, cost_per_second: 0.007407408 },
+                    { id: 'l40s', label: 'L40S', cost_per_hour: 4.44, cost_per_second: 0.001233333 },
+                ],
+            }];
+
+            context.preSubmitModal.hardwareProfile = 'h100-x4';
+            expect(context.getReplicateBaseHardwareCostDisplay()).toBe('$26.64/hr');
+            expect(context.getReplicateBaseHardwareCostDetail()).toBe('$0.007407/sec');
+
+            context.preSubmitModal.hardwareProfile = 'l40s';
+            expect(context.getReplicateBaseHardwareCostDisplay()).toBe('$4.44/hr');
+            expect(context.getReplicateBaseHardwareCostDetail()).toBe('$0.001233/sec');
+        });
     });
 
     describe('prepareJobPayload', () => {

tests/test_cloud_services.py

@@ -410,7 +410,7 @@ async def test_default_hardware_info(self) -> None:
         hardware = await get_hardware_info_async(self.store)
         self.assertEqual(hardware, DEFAULT_HARDWARE_INFO)
         self.assertIn("gpu-l40s", hardware)
-        self.assertIn("gpu-a100-large", hardware)
+        self.assertIn("gpu-h100", hardware)
 
     async def test_configured_hardware_info(self) -> None:
         """Test that configured hardware info overrides defaults."""
@@ -421,9 +421,8 @@ async def test_configured_hardware_info(self) -> None:
 
         # Configure custom hardware info
         custom_hardware = {
-            "gpu-a100-large": {"cost_per_second": 0.0014, "name": "A100 (80GB)"},
-            "gpu-h100": {"name": "H100 (80GB)", "cost_per_second": 0.0014},
-            "gpu-l40s": {"name": "L40S (48GB)", "cost_per_second": 0.000975},
+            "gpu-h100": {"name": "H100 (80GB)", "cost_per_second": 0.001525},
+            "gpu-l40s": {"name": "L40S (48GB)", "cost_per_second": 0.000972222},
         }
         await self.store.save_provider_config("replicate", {"hardware_info": custom_hardware})
 
@@ -445,8 +444,8 @@ async def test_default_hardware_cost_per_hour(self) -> None:
         clear_hardware_info_cache()
 
         cost = await get_default_hardware_cost_per_hour(self.store)
-        # L40S at $0.000975/sec = $3.51/hr
-        self.assertAlmostEqual(cost, 3.51, places=2)
+        # L40S at $0.000972222/sec = $3.50/hr
+        self.assertAlmostEqual(cost, 3.50, places=2)
 
     async def test_configured_hardware_cost_per_hour(self) -> None:
         """Test that configured hardware cost is used."""