fastevolve 0.3.3__tar.gz → 0.3.4__tar.gz

{fastevolve-0.3.3 → fastevolve-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastevolve
-Version: 0.3.3
+Version: 0.3.4
 Summary: Minimal open-source AlphaEvolve: LLM-driven program evolution with MAP-Elites islands, cascade evaluation, and a local Ollama ensemble.
 Project-URL: Homepage, https://github.com/tiagomonteiro0715/fastevolve
 Project-URL: Repository, https://github.com/tiagomonteiro0715/fastevolve
@@ -99,7 +99,7 @@ cfg = Config()
 cfg.iterations = 20
 cfg.checkpoint_path = "run.log"   # optional — resume if killed mid-run
 cfg.ensemble.models = [
-    ModelConfig(name="gemma3:e4b", provider="ollama", temperature=0.7, weight=1.0, role="fast"),
+    ModelConfig(name="gemma4:e4b", provider="ollama", temperature=0.7, weight=1.0, role="fast"),
 ]
 cfg.evaluator.cascade = [(correctness, 0.0)]
 
@@ -225,7 +225,7 @@ Start Ollama and pull the model first:
 
 ```bash
 ollama serve
-ollama pull gemma3:e4b
+ollama pull gemma4:e4b
 ```
 
 Then:
@@ -251,7 +251,7 @@ Then pick a `provider` per model in your config. You can freely mix providers in
 from fastevolve.llm_ensemble import ModelConfig
 
 cfg.ensemble.models = [
-    ModelConfig(name="gemma3:e4b", provider="ollama", temperature=0.6, weight=1.0, role="fast"),
+    ModelConfig(name="gemma4:e4b", provider="ollama", temperature=0.6, weight=1.0, role="fast"),
     ModelConfig(name="gpt-4o-mini", provider="openai", temperature=0.6, weight=1.0, role="fast"),
     ModelConfig(name="claude-opus-4-7", provider="anthropic", temperature=0.7, weight=1.0,
                 role="deep", options={"max_tokens": 4096}),

{fastevolve-0.3.3 → fastevolve-0.3.4}/README.md

@@ -69,7 +69,7 @@ cfg = Config()
 cfg.iterations = 20
 cfg.checkpoint_path = "run.log"   # optional — resume if killed mid-run
 cfg.ensemble.models = [
-    ModelConfig(name="gemma3:e4b", provider="ollama", temperature=0.7, weight=1.0, role="fast"),
+    ModelConfig(name="gemma4:e4b", provider="ollama", temperature=0.7, weight=1.0, role="fast"),
 ]
 cfg.evaluator.cascade = [(correctness, 0.0)]
 
@@ -195,7 +195,7 @@ Start Ollama and pull the model first:
 
 ```bash
 ollama serve
-ollama pull gemma3:e4b
+ollama pull gemma4:e4b
 ```
 
 Then:
@@ -221,7 +221,7 @@ Then pick a `provider` per model in your config. You can freely mix providers in
 from fastevolve.llm_ensemble import ModelConfig
 
 cfg.ensemble.models = [
-    ModelConfig(name="gemma3:e4b", provider="ollama", temperature=0.6, weight=1.0, role="fast"),
+    ModelConfig(name="gemma4:e4b", provider="ollama", temperature=0.6, weight=1.0, role="fast"),
     ModelConfig(name="gpt-4o-mini", provider="openai", temperature=0.6, weight=1.0, role="fast"),
     ModelConfig(name="claude-opus-4-7", provider="anthropic", temperature=0.7, weight=1.0,
                 role="deep", options={"max_tokens": 4096}),

{fastevolve-0.3.3 → fastevolve-0.3.4}/fastevolve/llm_ensemble/ollama.py

@@ -1,4 +1,5 @@
 import os
+import platform
 import shutil
 import subprocess
 import time
@@ -20,6 +21,34 @@ def _gpu_available() -> bool:
         return False
 
 
+@cache
+def _hardware_info() -> str:
+    if _gpu_available():
+        try:
+            r = subprocess.run(
+                ["nvidia-smi", "--query-gpu=name", "--format=csv,noheader"],
+                capture_output=True, text=True, timeout=2,
+            )
+            gpus = [g.strip() for g in r.stdout.strip().splitlines() if g.strip()]
+            if gpus:
+                tag = f"{gpus[0]}" + (f" ×{len(gpus)}" if len(gpus) > 1 else "")
+                return f"GPU: [bold green]{tag}[/]"
+        except Exception:
+            pass
+        return "GPU: [bold green]detected[/]"
+    cpu = platform.processor() or ""
+    if not cpu and os.path.exists("/proc/cpuinfo"):
+        try:
+            with open("/proc/cpuinfo") as f:
+                for line in f:
+                    if line.startswith("model name"):
+                        cpu = line.split(":", 1)[1].strip()
+                        break
+        except Exception:
+            pass
+    return f"CPU: [bold yellow]{cpu or 'unknown'}[/] ({os.cpu_count() or '?'} cores)"
+
+
 def start_ollama(host: str = "127.0.0.1:11434", *, wait: float = 5.0) -> None:
     """Start an ollama daemon with GPU-aware optimizations. No-op if one is already running."""
     for prefix in ("http://", "https://"):
@@ -39,9 +68,9 @@ def start_ollama(host: str = "127.0.0.1:11434", *, wait: float = 5.0) -> None:
         env.setdefault("OLLAMA_KV_CACHE_TYPE", "q8_0")
         env.setdefault("OLLAMA_NUM_PARALLEL", "4")
         env.setdefault("OLLAMA_MAX_LOADED_MODELS", "2")
-        log.info("[ollama] starting server in [bold]GPU[/] mode (flash_attn, q8_0 kv-cache, parallel=4, max_loaded=2)")
+        log.info("[ollama] starting server | %s | flash_attn, q8_0 kv-cache, parallel=4, max_loaded=2", _hardware_info())
     else:
-        log.info("[ollama] starting server in [bold]CPU[/] mode")
+        log.info("[ollama] starting server | %s", _hardware_info())
 
     path = shutil.which("ollama") or "/usr/local/bin/ollama"
     subprocess.Popen([path, "serve"], env=env,
@@ -60,7 +89,7 @@ class OllamaLLM(BaseLLM):
         except Exception:
             start_ollama(host=host)
             self.client = Client(host=host, timeout=timeout)
-        log.info("[ollama] %s → [bold]%s[/] mode", self.cfg.name, "GPU" if self._gpu else "CPU")
+        log.info("[ollama] %s | %s", self.cfg.name, _hardware_info())
         self._ensure_model()
 
     def _ensure_model(self):

{fastevolve-0.3.3 → fastevolve-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fastevolve"
-version = "0.3.3"
+version = "0.3.4"
 description = "Minimal open-source AlphaEvolve: LLM-driven program evolution with MAP-Elites islands, cascade evaluation, and a local Ollama ensemble."
 readme = "README.md"
 requires-python = ">=3.12"

{fastevolve-0.3.3 → fastevolve-0.3.4}/uv.lock

@@ -81,7 +81,7 @@ wheels = [
 
 [[package]]
 name = "fastevolve"
-version = "0.3.3"
+version = "0.3.4"
 source = { editable = "." }
 dependencies = [
     { name = "ollama" },