mirror of
https://github.com/elder-plinius/OBLITERATUS.git
synced 2026-05-02 07:55:11 +02:00
pliny
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
"""Obliteratus — Master Ablation Suite for HuggingFace transformers."""
|
|
|
|
__version__ = "0.1.2"
|
|
|
|
# Lazy imports for the main pipeline classes
|
|
__all__ = [
|
|
"AbliterationPipeline",
|
|
"InformedAbliterationPipeline",
|
|
"set_seed",
|
|
"run_sweep",
|
|
"SweepConfig",
|
|
"SweepResult",
|
|
"save_contribution",
|
|
"load_contributions",
|
|
"aggregate_results",
|
|
]
|
|
|
|
|
|
def __getattr__(name):
|
|
if name == "AbliterationPipeline":
|
|
from obliteratus.abliterate import AbliterationPipeline
|
|
return AbliterationPipeline
|
|
if name == "InformedAbliterationPipeline":
|
|
from obliteratus.informed_pipeline import InformedAbliterationPipeline
|
|
return InformedAbliterationPipeline
|
|
if name == "set_seed":
|
|
from obliteratus.reproducibility import set_seed
|
|
return set_seed
|
|
if name == "run_sweep":
|
|
from obliteratus.sweep import run_sweep
|
|
return run_sweep
|
|
if name == "SweepConfig":
|
|
from obliteratus.sweep import SweepConfig
|
|
return SweepConfig
|
|
if name == "SweepResult":
|
|
from obliteratus.sweep import SweepResult
|
|
return SweepResult
|
|
if name == "save_contribution":
|
|
from obliteratus.community import save_contribution
|
|
return save_contribution
|
|
if name == "load_contributions":
|
|
from obliteratus.community import load_contributions
|
|
return load_contributions
|
|
if name == "aggregate_results":
|
|
from obliteratus.community import aggregate_results
|
|
return aggregate_results
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|