biEzddlZddlZddlmZddlmZmZmZddl m Z e je Z GddeZy)N) CLIPConfigCLIPVisionModelWithProjectionPreTrainedModel)loggingc^eZdZeZdgZdeffd ZejddZ xZ S)IFSafetyCheckerCLIPEncoderLayerconfigct||t|j|_t j |jjd|_t j |jjd|_ y)N) super__init__r vision_config vision_modelnnLinearprojection_dimp_headw_head)selfr __class__s j/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/pipelines/deepfloyd_if/safety_checker.pyrzIFSafetyChecker.__init__s_  9&:N:NOii 4 4 C CQG ii 4 4 C CQG cx|j|d}|j|}|j}||kD}|j}t |rt j dt|D]-\}}|s tj||j||</|j|} | j} | |kD} | j} t | rt j dt| D]-\}} | s tj||j||</||| fS)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 rforwardzIFSafetyChecker.forwards@((4Q7  L1 %--/ % 3 %,,. }  NNB  $-]#; : C hhvc{'8'89s  :"[[6/779/+=/668 ! " NNB  )22D(E : $C$" hhvc{'8'89s  :}&888r)?r0) __name__ __module__ __qualname__r config_class_no_split_modulesrtorchno_gradr/ __classcell__)rs@rr r s;L+,HzHU]]_!9!9rr )numpyr"r6torch.nnr transformersrrrutilsr get_loggerr1rr rrr?s7 SS   H %/9o/9r