lyceum-cli 1.0.29__tar.gz → 1.0.30__tar.gz

{lyceum_cli-1.0.29 → lyceum_cli-1.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lyceum-cli
-Version: 1.0.29
+Version: 1.0.30
 Summary: Command-line interface for Lyceum Cloud Execution API
 Home-page: https://lyceum.technology
 Author: Lyceum Team

{lyceum_cli-1.0.29 → lyceum_cli-1.0.30}/lyceum/external/compute/execution/gpu_selection.py

@@ -44,6 +44,21 @@ GPU_DISPLAY_NAMES = {
     "gpu.rtx6000pro": "RTX 6000 Pro",
 }
 
+# VRAM in GB for each GPU profile (for showing excluded GPUs)
+GPU_VRAM_GB = {
+    "gpu": 16,
+    "gpu.t4": 16,
+    "gpu.t4.64gb": 16,
+    "gpu.a100": 40,
+    "gpu.a100.40gb": 40,
+    "gpu.a100.80gb": 80,
+    "gpu.h100": 80,
+    "gpu.h200": 141,
+    "gpu.l40s": 48,
+    "gpu.b200": 180,
+    "gpu.rtx6000pro": 48,
+}
+
 
 def format_gpu_name(profile: str) -> str:
     """Format GPU profile name for display."""
@@ -342,6 +357,7 @@ def display_results(data: dict, file_path: str | None = None) -> None:
 
         # Sort by VRAM size for better display
         sorted_configs = sorted(minimal_configs, key=lambda x: x.get("per_gpu_vram_gb", 0))
+        compatible_gpu_types = {cfg.get("gpu_type") for cfg in minimal_configs}
 
         for i, cfg in enumerate(sorted_configs):
             gpu_type = format_gpu_name(cfg.get("gpu_type", "?"))
@@ -362,6 +378,41 @@ def display_results(data: dict, file_path: str | None = None) -> None:
 
         console.print(gpu_table)
 
+        # Show excluded GPUs (those not in minimal_configs)
+        # Get total memory required from extraction result
+        mem_reqs = mem_config.get("memory_requirements", {})
+        total_mem_gb = 0
+        if mem_reqs:
+            # Sum up memory components
+            total_mem_gb = (
+                mem_reqs.get("model_weights", 0) +
+                mem_reqs.get("gradients", 0) +
+                mem_reqs.get("optimizer_states", 0) +
+                mem_reqs.get("activations", 0)
+            )
+
+        if not total_mem_gb and sorted_configs:
+            # Estimate from minimal config utilization
+            first_cfg = sorted_configs[0]
+            vram = first_cfg.get("per_gpu_vram_gb", 0)
+            util = first_cfg.get("vram_utilization_percent", 0)
+            if vram and util:
+                total_mem_gb = vram * (util / 100)
+
+        # Find GPUs that were excluded
+        excluded_gpus = []
+        for profile, vram in GPU_VRAM_GB.items():
+            if profile not in compatible_gpu_types and profile in ("gpu", "gpu.t4"):
+                # Only show common GPUs that users expect to see
+                excluded_gpus.append((profile, vram))
+
+        if excluded_gpus and total_mem_gb:
+            console.print()
+            console.print("[dim]Excluded GPUs (insufficient VRAM):[/dim]")
+            for profile, vram in sorted(excluded_gpus, key=lambda x: x[1]):
+                gpu_name = format_gpu_name(profile)
+                console.print(f"[dim]  • {gpu_name}: {vram} GB available, ~{total_mem_gb:.1f} GB required[/dim]")
+
     # Show run command hint if we have a best GPU and file path
     if best_gpu and file_path:
         machine_flag = best_gpu.get("gpu_type", "gpu")

{lyceum_cli-1.0.29 → lyceum_cli-1.0.30}/lyceum_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lyceum-cli
-Version: 1.0.29
+Version: 1.0.30
 Summary: Command-line interface for Lyceum Cloud Execution API
 Home-page: https://lyceum.technology
 Author: Lyceum Team

{lyceum_cli-1.0.29 → lyceum_cli-1.0.30}/setup.py

@@ -13,7 +13,7 @@ if readme_file.exists():
 
 setup(
     name="lyceum-cli",
-    version="1.0.29",
+    version="1.0.30",
     description="Command-line interface for Lyceum Cloud Execution API",
     long_description=long_description,
     long_description_content_type="text/markdown",