warm up model

img_nsfw.py

@@ -79,8 +79,8 @@ def check_nsfw(img, pipe):
     if torch.cuda.is_available():
         safety_checker_input = safety_checker_input.to('cuda')
     else:
-        safety_checker_input = safety_checker_input.half()
+        safety_checker_input = safety_checker_input
     from torch.cuda.amp import autocast
     with autocast():
-        _, nsfw_tags = pipe.safety_checker.forward(clip_input=safety_checker_input.pixel_values, images=img)
+        _, nsfw_tags = pipe.safety_checker.forward(clip_input=safety_checker_input.pixel_values.half(), images=img)
         return nsfw_tags