use generator device for get_score_autodan

Signed-off-by: Jeffrey Martin <[email protected]>
NVIDIA · Jul 22, 2024 · 3a0db34 · 3a0db34
1 parent 6c1bd1f
commit 3a0db34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/garak/resources/autodan/genetic.py b/garak/resources/autodan/genetic.py
@@ -594,7 +594,7 @@ def get_score_autodan(
     losses = []
     input_ids_list = []
     target_slices = []
-    device = generator.device if generator.device >= 0 else "cpu"
+    device = generator.device
     for item in test_controls:
         prefix_manager = AutoDanPrefixManager(
             generator=generator,