diff --git a/simpletuner/templates/partials/cloud_onboarding_flow.html b/simpletuner/templates/partials/cloud_onboarding_flow.html
index b445444e0..05fdcc1eb 100644
--- a/simpletuner/templates/partials/cloud_onboarding_flow.html
+++ b/simpletuner/templates/partials/cloud_onboarding_flow.html
@@ -224,11 +224,28 @@
Hardware
- L40S (48GB VRAM)
+
+
+
+
+
+
+
- Rate
- ~$3.50/hour
+ Cost (per hour)
+
+
+
+
Typical LoRA (2000 steps)
diff --git a/tests/js/job_submission.test.js b/tests/js/job_submission.test.js
index d81714808..ce5e9a1f2 100644
--- a/tests/js/job_submission.test.js
+++ b/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', () => {
diff --git a/tests/test_cloud_services.py b/tests/test_cloud_services.py
index f92e651db..d34aaecd9 100644
--- a/tests/test_cloud_services.py
+++ b/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."""
diff --git a/tests/test_cost_and_pricing.py b/tests/test_cost_and_pricing.py
index edf15a64d..2305d01e5 100644
--- a/tests/test_cost_and_pricing.py
+++ b/tests/test_cost_and_pricing.py
@@ -60,7 +60,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."""
@@ -88,8 +88,8 @@ async def test_default_hardware_cost_per_hour(self) -> None:
from simpletuner.simpletuner_sdk.server.services.cloud.replicate_client import get_default_hardware_cost_per_hour
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."""
diff --git a/tests/test_cost_estimation.py b/tests/test_cost_estimation.py
index 4314fa499..f11750d4e 100644
--- a/tests/test_cost_estimation.py
+++ b/tests/test_cost_estimation.py
@@ -49,12 +49,12 @@ def test_replicate_l40s_pricing(self):
hw = HardwareOption(
id="gpu-l40s",
name="L40S (48GB)",
- cost_per_second=0.000975,
+ cost_per_second=0.000972222,
memory_gb=48,
)
- # ~$3.51/hour
- self.assertAlmostEqual(hw.cost_per_hour, 3.51, places=2)
+ # ~$3.50/hour
+ self.assertAlmostEqual(hw.cost_per_hour, 3.50, places=2)
def test_to_dict(self):
"""Test to_dict includes all fields."""
@@ -226,7 +226,7 @@ def test_calculate_cost_one_hour(self):
default_hardware={
"gpu-l40s": {
"name": "L40S",
- "cost_per_second": 0.000975,
+ "cost_per_second": 0.000972222,
},
},
default_hardware_id="gpu-l40s",
@@ -234,7 +234,7 @@ def test_calculate_cost_one_hour(self):
# 1 hour = 3600 seconds
cost = provider.calculate_cost("gpu-l40s", 3600)
- self.assertAlmostEqual(cost, 3.51, places=2)
+ self.assertAlmostEqual(cost, 3.50, places=2)
def test_calculate_cost_unknown_hardware(self):
"""Test calculate_cost returns None for unknown hardware."""
@@ -285,7 +285,7 @@ def test_estimate_cost_bounds_accuracy(self):
default_hardware={
"gpu-l40s": {
"name": "L40S",
- "cost_per_second": 0.000975,
+ "cost_per_second": 0.000972222,
},
},
default_hardware_id="gpu-l40s",
@@ -294,8 +294,8 @@ def test_estimate_cost_bounds_accuracy(self):
# Typical training job: 30 minutes
estimate = provider.estimate_cost("gpu-l40s", 30)
- # Estimated cost: 30 min * $0.000975 * 60 = $1.755
- self.assertAlmostEqual(estimate["estimated"], 1.755, places=2)
+ # Estimated cost: 30 min * $0.000972222 * 60 = $1.75
+ self.assertAlmostEqual(estimate["estimated"], 1.75, places=2)
# Bounds check
self.assertLess(estimate["min"], estimate["estimated"])
@@ -331,7 +331,7 @@ def test_estimate_long_job(self):
default_hardware={
"gpu-l40s": {
"name": "L40S",
- "cost_per_second": 0.000975,
+ "cost_per_second": 0.000972222,
},
},
default_hardware_id="gpu-l40s",
@@ -340,8 +340,8 @@ def test_estimate_long_job(self):
# 24 hours = 1440 minutes
estimate = provider.estimate_cost("gpu-l40s", 1440)
- # Expected: 1440 * 60 * 0.000975 = $84.24
- self.assertAlmostEqual(estimate["estimated"], 84.24, places=2)
+ # Expected: 1440 * 60 * 0.000972222 = $84.00
+ self.assertAlmostEqual(estimate["estimated"], 84.00, places=2)
class TestReplicatePricing(unittest.TestCase):
@@ -366,20 +366,20 @@ def test_replicate_l40s_pricing(self):
self.assertIsNotNone(hw)
self.assertEqual(hw.memory_gb, 48)
- # L40S costs ~$0.000975/s = ~$3.51/hour
- self.assertAlmostEqual(hw.cost_per_hour, 3.51, places=2)
+ # L40S costs ~$0.000972222/s = ~$3.50/hour
+ self.assertAlmostEqual(hw.cost_per_hour, 3.50, places=2)
- def test_replicate_a100_pricing(self):
- """Test Replicate A100 pricing."""
+ def test_replicate_h100_pricing(self):
+ """Test Replicate H100 pricing."""
from simpletuner.simpletuner_sdk.server.services.cloud.pricing import get_pricing_provider
provider = get_pricing_provider("replicate")
- hw = provider.get_hardware_by_id("gpu-a100-large")
+ hw = provider.get_hardware_by_id("gpu-h100")
self.assertIsNotNone(hw)
self.assertEqual(hw.memory_gb, 80)
- # A100 costs ~$0.0014/s = ~$5.04/hour
- self.assertAlmostEqual(hw.cost_per_hour, 5.04, places=2)
+ # H100 costs ~$0.001525/s = ~$5.49/hour
+ self.assertAlmostEqual(hw.cost_per_hour, 5.49, places=2)
def test_replicate_default_hardware(self):
"""Test Replicate default hardware is L40S."""
diff --git a/tests/test_replicate_hardware_profiles.py b/tests/test_replicate_hardware_profiles.py
index 881278582..e8f53d0b3 100644
--- a/tests/test_replicate_hardware_profiles.py
+++ b/tests/test_replicate_hardware_profiles.py
@@ -25,6 +25,12 @@ def test_invalid_profile_raises(self):
with self.assertRaises(ValueError):
normalize_replicate_hardware_profile("a10g")
+ def test_default_hardware_info_includes_h100_and_l40s_pricing(self):
+ from simpletuner.simpletuner_sdk.server.services.cloud.replicate_client import DEFAULT_HARDWARE_INFO
+
+ self.assertEqual(DEFAULT_HARDWARE_INFO["gpu-l40s"]["cost_per_second"], 0.000972222)
+ self.assertEqual(DEFAULT_HARDWARE_INFO["gpu-h100"]["cost_per_second"], 0.001525)
+
async def test_replicate_client_uses_profile_model_for_latest_version(self):
from simpletuner.simpletuner_sdk.server.services.cloud.replicate_client import ReplicateCogClient
diff --git a/tests/test_webui_e2e.py b/tests/test_webui_e2e.py
index 1317d5aa4..356f27558 100644
--- a/tests/test_webui_e2e.py
+++ b/tests/test_webui_e2e.py
@@ -1770,9 +1770,11 @@ def _install_cloud_modal_harness(self, driver) -> bool:
name: 'Replicate',
hardware_profile: 'l40s-x4',
hardware_profiles: [
- { id: 'h100', label: 'H100', hardware_type: 'H100' },
- { id: 'h100-x8', label: '8x H100', hardware_type: '8x H100' },
- { id: 'l40s-x4', label: '4x L40S', hardware_type: '4x L40S' }
+ { id: 'h100', label: 'H100', hardware_type: 'H100', cost_per_hour: 5.49, cost_per_second: 0.001525 },
+ { id: 'h100-x8', label: '8x H100', hardware_type: '8x H100', cost_per_hour: 43.92, cost_per_second: 0.0122 },
+ { id: 'l40s', label: 'L40S', hardware_type: 'L40S', cost_per_hour: 3.50, cost_per_second: 0.000972222 },
+ { id: 'l40s-x4', label: '4x L40S', hardware_type: '4x L40S', cost_per_hour: 14.00, cost_per_second: 0.003888888 },
+ { id: 'l40s-x8', label: '8x L40S', hardware_type: '8x L40S', cost_per_hour: 28.00, cost_per_second: 0.007777776 }
]
}];
comp.providerConfig = { config: { hardware_profile: 'l40s-x4' }, cost_limit_enabled: false };
@@ -1886,6 +1888,80 @@ def scenario(driver, _browser):
self.for_each_browser("test_replicate_hardware_profile_persists_and_submits", scenario)
+ def test_replicate_settings_hardware_buttons_update_cost_and_persist(self) -> None:
+ self.with_sample_environment()
+
+ def scenario(driver, _browser):
+ self._open_cloud_tab(driver)
+ driver.execute_script("localStorage.setItem('cloud_replicate_hardware_profile', 'h100-x8');")
+ self.assertTrue(self._install_cloud_modal_harness(driver))
+
+ driver.execute_script(
+ """
+ const comp = Alpine.$data(document.querySelector('#cloud-tab-content'));
+ comp.preSubmitModal.hardwareProfile = 'h100-x8';
+ comp.showSettingsPanel = true;
+ """
+ )
+
+ WebDriverWait(driver, 5).until(
+ lambda d: d.execute_script(
+ "return Array.from(document.querySelectorAll('[data-testid=\"cloud-settings-hardware-profile\"]'))"
+ ".filter((button) => button.offsetParent !== null).length === 2;"
+ )
+ )
+ WebDriverWait(driver, 5).until(
+ lambda d: d.execute_script(
+ "const panel = document.querySelector('.cloud-settings-panel');"
+ "return panel && panel.innerText.includes('$43.92/hr') && "
+ "panel.innerText.includes('$0.012200/sec');"
+ )
+ )
+
+ pressed_before = driver.execute_script(
+ """
+ return Array.from(document.querySelectorAll('[data-testid="cloud-settings-hardware-profile"]'))
+ .filter((button) => button.offsetParent !== null)
+ .map((button) => ({ text: button.innerText.trim(), pressed: button.getAttribute('aria-pressed') }));
+ """
+ )
+ self.assertIn({"text": "H100", "pressed": "true"}, pressed_before)
+ self.assertIn({"text": "L40S", "pressed": "false"}, pressed_before)
+
+ clicked = driver.execute_script(
+ """
+ const button = Array.from(document.querySelectorAll('[data-testid="cloud-settings-hardware-profile"]'))
+ .find((candidate) => candidate.offsetParent !== null && candidate.innerText.trim() === 'L40S');
+ if (!button) { return false; }
+ button.click();
+ return true;
+ """
+ )
+ self.assertTrue(clicked)
+
+ WebDriverWait(driver, 5).until(
+ lambda d: d.execute_script("return localStorage.getItem('cloud_replicate_hardware_profile');") == "l40s-x8"
+ )
+ WebDriverWait(driver, 5).until(
+ lambda d: d.execute_script(
+ "const panel = document.querySelector('.cloud-settings-panel');"
+ "return panel && panel.innerText.includes('$28.00/hr') && "
+ "panel.innerText.includes('$0.007778/sec');"
+ )
+ )
+
+ pressed_after = driver.execute_script(
+ """
+ return Array.from(document.querySelectorAll('[data-testid="cloud-settings-hardware-profile"]'))
+ .filter((button) => button.offsetParent !== null)
+ .map((button) => ({ text: button.innerText.trim(), pressed: button.getAttribute('aria-pressed') }));
+ """
+ )
+ self.assertIn({"text": "H100", "pressed": "false"}, pressed_after)
+ self.assertIn({"text": "L40S", "pressed": "true"}, pressed_after)
+
+ self.for_each_browser("test_replicate_settings_hardware_buttons_update_cost_and_persist", scenario)
+
class CloudUploadStatusTestCase(_TrainerPageMixin, WebUITestCase):
"""Ensure uploading jobs appear in the cloud job list."""