mirror of
https://github.com/elder-plinius/OBLITERATUS.git
synced 2026-07-12 07:06:32 +02:00
pliny
16 lines
581 B
Python
16 lines
581 B
Python
from obliteratus.strategies.registry import STRATEGY_REGISTRY, register_strategy, get_strategy
|
|
from obliteratus.strategies.layer_removal import LayerRemovalStrategy
|
|
from obliteratus.strategies.head_pruning import HeadPruningStrategy
|
|
from obliteratus.strategies.ffn_ablation import FFNAblationStrategy
|
|
from obliteratus.strategies.embedding_ablation import EmbeddingAblationStrategy
|
|
|
|
__all__ = [
|
|
"STRATEGY_REGISTRY",
|
|
"register_strategy",
|
|
"get_strategy",
|
|
"LayerRemovalStrategy",
|
|
"HeadPruningStrategy",
|
|
"FFNAblationStrategy",
|
|
"EmbeddingAblationStrategy",
|
|
]
|