Function replace_target_caption

scripts/eval_step_attack.py:191–199

(obs_text: str, som_id: Union[int, List[int]], target_caption: str)

def replace_target_caption(obs_text: str, som_id: Union[int, List[int]], target_caption: str) -> str:
    if isinstance(som_id, list):
        modified_obs_text = obs_text
        for idx in som_id:
            modified_obs_text = _replace_target_caption(modified_obs_text, idx, target_caption)
        print(f"Modified observation text: {modified_obs_text}")
        return modified_obs_text
    else:
        return _replace_target_caption(obs_text, som_id, target_caption)


def load(example: dict, attack: str, use_caption: bool, args) -> tuple: