commit

app.py

@@ -133,7 +133,7 @@ def disabled_safety_checker(images, clip_input):
 
 image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=torch.bfloat16).to(device)
 image_pipe.enable_model_cpu_offload()
-image_pipe.safety_checker = disabled_safety_checker
+image_pipe.safety_checker = None
 
 video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     "THUDM/CogVideoX-5b-I2V",
@@ -141,7 +141,7 @@ video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
 ).to(device)
 video_pipe.vae.enable_tiling()
 video_pipe.vae.enable_slicing()
-video_pipe.safety_checker = disabled_safety_checker
+video_pipe.safety_checker = None
 
 # functionality