CalvinBackup/OBLITERATUS
1
0
Fork 0
mirror of https://github.com/elder-plinius/OBLITERATUS.git synced 2026-08-17 16:37:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

ci: enforce Ruff across all scripts

Browse Source
This commit is contained in:
Joseph Magly
2026-08-16 03:49:09 -04:00
parent aa182cc448
commit 3574800094
12 changed files with 19 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/gemma4_gradient_aspa.py
+1 -2
View File
@@ -9,7 +9,6 @@ This should let us recover MORE MMLU than uniform blending while
keeping refusals at absolute zero.
"""
import argparse
import gc
import json
import time
@@ -197,7 +196,7 @@ def main():
out_dir = Path("runs/gemma4-12b-surgery/gradient_aspa")
out_dir.mkdir(parents=True, exist_ok=True)
print(f"Loading tokenizer and v1 model...", flush=True)
print("Loading tokenizer and v1 model...", flush=True)
tok = AutoTokenizer.from_pretrained(v1_model_path, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
v1_model_path, torch_dtype=torch.bfloat16, trust_remote_code=True