attacklm 0.6.7__tar.gz → 0.6.8__tar.gz

{attacklm-0.6.7 → attacklm-0.6.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: attacklm
-Version: 0.6.7
+Version: 0.6.8
 Summary: QLoRA-fine-tuned Qwen2.5-Coder for offensive/defensive security and AI red-teaming
 Project-URL: Homepage, https://github.com/Veedubin/AttackLM
 Project-URL: Repository, https://github.com/Veedubin/AttackLM

{attacklm-0.6.7 → attacklm-0.6.8}/scripts/train_template.py

@@ -3366,19 +3366,19 @@ def main() -> None:
             except (TypeError, ValueError):
                 loss_val = 0.0
 
-            # --- VRAM (allocated + cached / total) ---
+            # --- VRAM (used/total with alloc+cache breakdown) ---
             vram_str = ""
             if is_cuda():
                 try:
                     allocated_gb = gpu_mem_allocated_bytes() / (1024**3)
                     cached_gb = gpu_mem_cached_bytes() / (1024**3)
+                    used_gb = allocated_gb + cached_gb
                     total_gb = torch.cuda.get_device_properties(0).total_memory / (
                         1024**3
                     )
                     vram_str = (
-                        f"VRAM alloc {allocated_gb:.1f} "
-                        f"cache {cached_gb:.1f} "
-                        f"/{total_gb:.1f} GB"
+                        f"VRAM {used_gb:.1f}/{total_gb:.1f} GB"
+                        f" ({allocated_gb:.1f} alloc + {cached_gb:.1f} cache)"
                     )
                 except Exception:
                     pass

attacklm-0.6.8/src/attacklm/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.6.8"

attacklm-0.6.7/src/attacklm/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.6.7"