+hEzSSKrSSKrSSKJr SSKJrJrJr SSK J r \ R"\ 5r "SS\5rg)N) CLIPConfigCLIPVisionModelWithProjectionPreTrainedModel)loggingcl^\rSrSr\rS/rS\4U4Sjjr\R"5SSj5r Sr U=r $)IFSafetyChecker CLIPEncoderLayerconfigc>[TU]U5 [UR5Ul[ R "URRS5Ul[ R "URRS5Ul g)N) super__init__r vision_config vision_modelnnLinearprojection_dimp_headw_head)selfr __class__s i/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/deepfloyd_if/safety_checker.pyrIFSafetyChecker.__init__s_  9&:N:NOii 4 4 C CQG ii 4 4 C CQG cURU5SnURU5nUR5nXc:nUR5n[ U5(a[ R S5 [U5H2upxU(dM[R"X'R5X''M4 URU5n U R5n X:n U R5n [ U 5(a[ R S5 [U 5H2upzU (dM[R"X'R5X''M4 X&U 4$)NrzPotential NSFW content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed.zPotential watermarked content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed.) rrflattentolistanyloggerwarning enumeratenpzerosshaper) r clip_inputimages p_threshold w_threshold image_embeds nsfw_detectedidxnsfw_detected_watermark_detectedwatermark_detected_s rforwardIFSafetyChecker.forwards/((4Q7  L1 %--/ %3 %,,. }   NNB  $-]#; C~ hhv{'8'89 $<"[[6/779/=/668 ! " " NNB  )22D(E $C"" hhv{'8'89 )F&888r)rrr)?r3) __name__ __module__ __qualname____firstlineno__r config_class_no_split_modulesrtorchno_gradr1__static_attributes__ __classcell__)rs@rr r s9L+,HzH ]]_!9!9rr )numpyr$r:torch.nnr transformersrrrutilsr get_loggerr4r!r rrrDs5 SS   H %/9o/9r