U \}}t|tjr>|||dq|||dqi}t|rr|jj|d|d}t|r|jj|dd} | j|d<||d<t |S)Nrrpt)return_tensorsr-)r/rr ) r$ isinstancerr%lenr tokenizerZimage_processorrdict) rZtextsr-ZimagesZ texts_valuesr r+dataencodingr*rrrtokenize6s"      zCLIPModel.tokenizecCs|jSr )rrrrrr2NszCLIPModel.tokenizer) output_pathr cCs|j||j|dSr )rZsave_pretrainedr)rr7rrrsaveRs zCLIPModel.save) input_pathr cCs t|dS)N)r)r)r9rrrloadVszCLIPModel.load)rN)T)__name__ __module__ __qualname__strr rrr'ZTensorr,rboolr6propertyr rr2r8 staticmethodr: __classcell__rrrrr s $ $r) typingrrr'r ZPILrrModulerrrrrs