Update test.py

xh-liu · Aug 27, 2021 · 820b347 · 820b347
1 parent b2ff6cd
commit 820b347
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test.py b/test.py
@@ -24,8 +24,6 @@
 
 ori_cap = opt.ori_cap.split()
 new_cap = opt.new_cap.split()
-import pdb
-pdb.set_trace()
 global_edit = False
 
 alpha = 5
@@ -88,7 +86,7 @@
 
 # manipulate without optimizing perturbations
 manipulated_ori = open_edit_optimizer.open_edit_model(data, mode='manipulate', ori_cap=ori_txt, new_cap=new_txt, alpha=alpha)
-visuals['manipulated_ori'] = manipulated_ori[0][0]
+visuals['manipulated_ori'] = manipulated_ori[0]
 
 # optimize perturbations
 for iter_cnt in range(optimize_iter):