Add LoKR/LoRA config debug output to tier 3 benchmark

Author: CalamitousFelicitousness

benchmark_lokr.py

@@ -246,6 +246,12 @@ def weight_space_svd(lokr_path, lokr_name, ranks, no_offload=False):
         t0 = time.time()
         lora_config, lora_sd = convert_to_lora(pipe.transformer, rank, adapter_name=adapter_name, progressbar=True)
         print(f"  Converted in {time.time() - t0:.1f}s")
+        print(f"  LoRA config: alpha={lora_config.lora_alpha}, r={lora_config.r}")
+
+        # Also print the LoKR config for reference
+        lokr_cfg = pipe.transformer.peft_config.get(adapter_name)
+        if lokr_cfg:
+            print(f"  LoKR config: alpha={lokr_cfg.alpha}, r={lokr_cfg.r}")
 
         # Compare each module: LoKR delta vs LoRA delta (lora_B @ lora_A)
         results = []
@@ -343,10 +349,16 @@ def benchmark_tier3_svd(pipe, prompt, seed, rank, lokr_path, lokr_name):
     adapter_name = next(iter(pipe.transformer.peft_config.keys()))
     print(f"  Adapter name: {adapter_name}")
 
+    t0 = time.time()
+    lokr_cfg = pipe.transformer.peft_config.get(adapter_name)
+    if lokr_cfg:
+        print(f"  LoKR config: alpha={lokr_cfg.alpha}, r={lokr_cfg.r}")
+
     t0 = time.time()
     lora_config, lora_sd = convert_to_lora(pipe.transformer, rank, adapter_name=adapter_name, progressbar=True)
     convert_time = time.time() - t0
     print(f"  Loaded LoKR in {load_time:.1f}s, converted to LoRA in {convert_time:.1f}s")
+    print(f"  LoRA config: alpha={lora_config.lora_alpha}, r={lora_config.r}")
 
     pipe.transformer.delete_adapters(adapter_name)
     inject_adapter_in_model(lora_config, pipe.transformer, adapter_name=adapter_name)