remove checker during evaluation

Tianwei Yin · Tianwei Yin · commit 71fed37def3c · 2023-12-08T19:14:12.000-08:00
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)
 
