Lakesenberg commited on
Commit
8946807
·
verified ·
1 Parent(s): b07b32f

Upload folder using huggingface_hub

Browse files
covariances.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c44e9775b693046a38ef2b46c741db221983f6060f2f0129b27c7cfebe03ce68
3
+ size 86936387582
nemotron-adapter-aligned.tar.gz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21dcc3c6c9d262d7158ac9e3d01f40c433b3f8c462615e26f7371addfe50a7bf
3
+ size 3220606796
nemotron-adapter-aligned/adapter_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "peft_type": "LORA",
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/home/lakesenberg/models/nemotron-3-nano-30b-a3b-bf16",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "lora_alpha": 32,
10
+ "lora_dropout": 0.0,
11
+ "modules_to_save": null,
12
+ "r": 32,
13
+ "rank_pattern": {},
14
+ "alpha_pattern": {},
15
+ "target_modules": [
16
+ "down_proj",
17
+ "in_proj",
18
+ "k_proj",
19
+ "o_proj",
20
+ "out_proj",
21
+ "q_proj",
22
+ "up_proj",
23
+ "v_proj"
24
+ ],
25
+ "task_type": "CAUSAL_LM"
26
+ }
nemotron-adapter-aligned/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:007f3fce94c9d4cbbe2f1686d2cd25c51af9f0689a65f1e477ee5d968dc655f8
3
+ size 3537263096
run.log ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ === [Sun May 24 13:19:04 UTC 2026] AA-SVD H200 MODE=smoke ===
2
+ `torch_dtype` is deprecated! Use `dtype` instead!
3
+ The fast path is not available because on of `(selective_state_update, causal_conv1d_fn, causal_conv1d_update)` is None. Falling back to the naive implementation. To install follow https://github.com/state-spaces/mamba/#installation and https://github.com/Dao-AILab/causal-conv1d
4
+ === AA-SVD H200 run ===
5
+ Config(base_model_id='/home/lakesenberg/models/nemotron-3-nano-30b-a3b-bf16', tinker_adapter_path='/home/lakesenberg/nemotron_lora_work/adapter_v2/final', aligned_output_path='/home/lakesenberg/aa_svd_work/nemotron-adapter-aligned', plain_output_path='/home/lakesenberg/aa_svd_work/nemotron-adapter-plain', covariance_path='/home/lakesenberg/aa_svd_work/covariances.pt', submission_zip_path='/home/lakesenberg/aa_svd_work/submission', use_precomputed_covariance=False, calibration_prompts_path=None, num_calibration_samples=8, max_calibration_seq_len=1024, calibration_batch_size=1, base_model_load_in_4bit=False, base_model_device_map='auto', forced_fused_rank=32, svd_damping=1e-05, use_double_precision_svd=True, fallback_to_plain_svd=True, enable_activation_aware_for=('in_proj', 'qkv_proj', 'gate_up_proj'), device='cuda', seed=0, skip_plain_build=True, skip_smoke_gen=True)
6
+ Using built-in default calibration prompts.
7
+ Calibration prompts ready: 8
8
+ Loading base model /home/lakesenberg/models/nemotron-3-nano-30b-a3b-bf16 (4bit=False) ...
9
+
10
+ after base load GPU mem: used=58.8 GiB, peak=58.8 GiB
11
+ Traceback (most recent call last):
12
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 600, in <module>
13
+ main()
14
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 548, in main
15
+ covariances = collect_activation_covariance(cfg, calib)
16
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
17
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 327, in collect_activation_covariance
18
+ accums[weight_key] = _CovAccumulator(module.in_features, device=device)
19
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
20
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 278, in __init__
21
+ self.cov = torch.zeros(in_dim, in_dim, dtype=torch.float32, device=device)
22
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
23
+ torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 28.00 MiB. GPU 0 has a total capacity of 139.72 GiB of which 4.69 MiB is free. Including non-PyTorch memory, this process has 139.71 GiB memory in use. Of the allocated memory 139.12 GiB is allocated by PyTorch, and 146.50 KiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)
24
+ === [Sun May 24 13:19:36 UTC 2026] AA-SVD H200 MODE=smoke ===
25
+ `torch_dtype` is deprecated! Use `dtype` instead!
26
+ The fast path is not available because on of `(selective_state_update, causal_conv1d_fn, causal_conv1d_update)` is None. Falling back to the naive implementation. To install follow https://github.com/state-spaces/mamba/#installation and https://github.com/Dao-AILab/causal-conv1d
27
+ === AA-SVD H200 run ===
28
+ Config(base_model_id='/home/lakesenberg/models/nemotron-3-nano-30b-a3b-bf16', tinker_adapter_path='/home/lakesenberg/nemotron_lora_work/adapter_v2/final', aligned_output_path='/home/lakesenberg/aa_svd_work/nemotron-adapter-aligned', plain_output_path='/home/lakesenberg/aa_svd_work/nemotron-adapter-plain', covariance_path='/home/lakesenberg/aa_svd_work/covariances.pt', submission_zip_path='/home/lakesenberg/aa_svd_work/submission', use_precomputed_covariance=False, calibration_prompts_path=None, num_calibration_samples=8, max_calibration_seq_len=1024, calibration_batch_size=1, base_model_load_in_4bit=False, base_model_device_map='auto', forced_fused_rank=32, svd_damping=1e-05, use_double_precision_svd=True, fallback_to_plain_svd=True, enable_activation_aware_for=('in_proj', 'qkv_proj', 'gate_up_proj'), device='cuda', seed=0, skip_plain_build=True, skip_smoke_gen=True)
29
+ Using built-in default calibration prompts.
30
+ Calibration prompts ready: 8
31
+ Loading base model /home/lakesenberg/models/nemotron-3-nano-30b-a3b-bf16 (4bit=False) ...
32
+
33
+ `use_return_dict` is deprecated! Use `return_dict` instead!
34
+ MoE expert LoRA serving for nemotron models is experimental in vLLM and not yet supported in SGLang. The adapter will be produced but may not work with all serving configurations.
35
+ after base load GPU mem: used=58.8 GiB, peak=58.8 GiB
36
+ Registered hooks on 3008 fused-target modules.
37
+ step 0: GPU mem: used=58.9 GiB, peak=59.7 GiB
38
+ [covariance collection] took 54.07s
39
+ Saved 81.0 GiB -> /home/lakesenberg/aa_svd_work/covariances.pt
40
+ Have 3008 covariance matrices.
41
+ patched: patched_merge_fused_projections
42
+ Traceback (most recent call last):
43
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 601, in <module>
44
+ main()
45
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 557, in main
46
+ build_adapter(cfg, cfg.aligned_output_path, "activation-aware")
47
+ File "/home/lakesenberg/code/nemotron_lora_train/run_aa_svd_h200.py", line 445, in build_adapter
48
+ weights.build_lora_adapter(
49
+ File "/home/lakesenberg/miniforge3/envs/nemo/lib/python3.11/site-packages/tinker_cookbook/weights/_adapter.py", line 150, in build_lora_adapter
50
+ peft_weights, target_modules, rank_overrides = _convert_adapter(
51
+ ^^^^^^^^^^^^^^^^^
52
+ File "/home/lakesenberg/miniforge3/envs/nemo/lib/python3.11/site-packages/tinker_cookbook/weights/_adapter.py", line 293, in _convert_adapter
53
+ _expand_expert_weights(
54
+ File "/home/lakesenberg/miniforge3/envs/nemo/lib/python3.11/site-packages/tinker_cookbook/weights/_adapter.py", line 506, in _expand_expert_weights
55
+ raise WeightsAdapterError(
56
+ tinker_cookbook.exceptions.WeightsAdapterError: Expert LoRA weights must be 3D, got lora_A: torch.Size([32, 1856]), lora_B: torch.Size([2688, 32]) for key targeting 'backbone.layers.1.mixer.experts.0.down_proj.weight'
57
+ === [Sun May 24 13:24:53 UTC 2026] AA-SVD H200 MODE=smoke ===
58
+ MoE expert LoRA serving for nemotron models is experimental in vLLM and not yet supported in SGLang. The adapter will be produced but may not work with all serving configurations.
59
+ === AA-SVD H200 run ===
60
+ Config(base_model_id='/home/lakesenberg/models/nemotron-3-nano-30b-a3b-bf16', tinker_adapter_path='/home/lakesenberg/nemotron_lora_work/adapter_v2/final', aligned_output_path='/home/lakesenberg/aa_svd_work/nemotron-adapter-aligned', plain_output_path='/home/lakesenberg/aa_svd_work/nemotron-adapter-plain', covariance_path='/home/lakesenberg/aa_svd_work/covariances.pt', submission_zip_path='/home/lakesenberg/aa_svd_work/submission', use_precomputed_covariance=True, calibration_prompts_path=None, num_calibration_samples=8, max_calibration_seq_len=1024, calibration_batch_size=1, base_model_load_in_4bit=False, base_model_device_map='auto', forced_fused_rank=32, svd_damping=1e-05, use_double_precision_svd=True, fallback_to_plain_svd=True, enable_activation_aware_for=('in_proj', 'qkv_proj', 'gate_up_proj'), device='cuda', seed=0, skip_plain_build=True, skip_smoke_gen=True)
61
+ Using built-in default calibration prompts.
62
+ Calibration prompts ready: 8
63
+ Loading precomputed covariances from /home/lakesenberg/aa_svd_work/covariances.pt
64
+ Have 3008 covariance matrices.
65
+ patched: patched_merge_fused_projections
66
+ [activation-aware adapter build] took 6.29s
67
+ activation-aware build wrote 2 files -> /home/lakesenberg/aa_svd_work/nemotron-adapter-aligned
68
+
69
+ === DONE ===
70
+ Aligned adapter : /home/lakesenberg/aa_svd_work/nemotron-adapter-aligned
71
+ Submission zip : /home/lakesenberg/aa_svd_work/submission.zip (3.0 GiB)
72
+
73
+ Kaggle tips (RTX Pro 6000):
74
+ - set CFG.base_model_load_in_4bit = True
75
+ - use this notebook with adapter uploaded as Kaggle Model dataset
76
+ === [Sun May 24 13:28:09 UTC 2026] AA-SVD finished ===
submission.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8932e71b0d190c4ddd861afe9cb6d90e4cb7dedc47b4a611a134054fba605ae1
3
+ size 3221191769