warm up model

img_nsfw.py

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