remove checker during evaluation

mit-han-lab · Dec 9, 2023 · 71fed37 · 71fed37
1 parent 34c3b59
commit 71fed37
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/evaluation/single_object/run.py b/evaluation/single_object/run.py
@@ -21,6 +21,7 @@ def main():
         set_seed(args.seed)
 
     model = convert_model_to_pipeline(args, accelerator.device)
+    model.safety_checker = None 
 
     os.makedirs(args.output_dir, exist_ok=True)