Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/tamperbench/whitebox/attacks/gcg/gcg.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
from typing_extensions import override

from tamperbench.whitebox.attacks.base import TamperAttack, TamperAttackConfig
from tamperbench.whitebox.attacks.gcg.implementation import GCGConfig
from tamperbench.whitebox.attacks.registry import register_attack
from tamperbench.whitebox.evals.gcg.gcg import GCGEvaluation, GCGEvaluationConfig
from tamperbench.whitebox.evals.gcg.implementation import GCGConfig
from tamperbench.whitebox.evals.output_schema import EvaluationSchema
from tamperbench.whitebox.utils.names import AttackName, EvalName

Expand Down
2 changes: 1 addition & 1 deletion src/tamperbench/whitebox/evals/gcg/gcg.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@
from tqdm import tqdm
from typing_extensions import override

from tamperbench.whitebox.attacks.gcg.implementation import GCG, GCGConfig
from tamperbench.whitebox.evals.base import load_model_and_tokenizer
from tamperbench.whitebox.evals.gcg.implementation import GCG, GCGConfig
from tamperbench.whitebox.evals.output_schema import InferenceSchema
from tamperbench.whitebox.evals.registry import register_evaluation
from tamperbench.whitebox.evals.strong_reject.strong_reject import (
Expand Down
2 changes: 1 addition & 1 deletion tests/attacks/test_gcg_attack.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from dotenv import load_dotenv

from tamperbench.whitebox.attacks.gcg.gcg import GCGAttack, GCGAttackConfig
from tamperbench.whitebox.attacks.gcg.implementation import GCGConfig
from tamperbench.whitebox.evals.gcg.implementation import GCGConfig
from tamperbench.whitebox.evals.output_schema import EvaluationSchema
from tamperbench.whitebox.utils.models.config import ModelConfig
from tamperbench.whitebox.utils.names import EvalName, MetricName
Expand Down