,j SSKJr SSKJr SSKJrJrJrJrJ r SSK r SSK J s J r SrS S\\S\\ \\4S\\S \ \\ R$4S \4 S jjrSS\\S\\ \\4S \ \\ R$4S \4S jjrg))partial)islice)CallableListOptionalSequenceUnionNc#f# [U5n[[X!55nU(dgUv M"7f)zBatch data into lists of length *n*. The last batch may be shorter. NOTE based on more-itertools impl, to be replaced by python 3.12 itertools.batched impl N)iterlistr)iterablenitbatchs ڬC:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\open_clip/zero_shot_classifier.pybatchedr s1 hB VB]#  s/1 classnames templatesnum_classes_per_batchdeviceuse_tqdmc~^^^^^^[T[5(a[T5S:de[U[5(a[U5S:de[TS[5m[T5m[U5nU(a)SSKnUcSO US- U-S-n [ URXS9n O[ n UUUUUU4Sjn [R"5 U(a<U "[X$55V s/sH o"U 5PM n n [R"U SS9nOU "U5nSSS5 U$s sn f!,(df  W$=f)aBuild zero-shot classifier weights by iterating over class names in batches Args: model: CLIP model instance tokenizer: CLIP tokenizer instance classnames: A sequence of class (label) names templates: A sequence of callables or format() friendly strings to produce templates per class name num_classes_per_batch: The number of classes to batch together in each forward, all if None device: Device to use. use_tqdm: Enable TQDM progress bar. rN)total unit_scalecl>[U5nUVVs/sH-nT H#nT (aURU5OU"U5PM% M/ nnnT "U5RT5nTRUSS9nUR UTS5R SS9nXUR SSS9- nURnU$s snnf)NT) normalizerdim)r keepdim)lenformatto encode_textreshapemeannormT) batch_classnamesnum_batch_classesctemplatetextsclass_embeddingsrmodel num_templatesr tokenizer use_formats r_process_batch2build_zero_shot_classifier.._process_batch4s 01L\wL\qmvaiz#x{BmvBL\w% ##F+ ,,Ud,C+334E}VXY^^cd^e+.C.CSW.C.XX+-- xs4B0r) isinstancerr"strtqdmrr torchno_gradrcat)r0r2rrrrr num_classesr8num_iter iter_wrapr4rbatched_embedszeroshot_weightsr1r3s`` ` ` @@rbuild_zero_shot_classifierrAs& i * *s9~/AA A j( + +J!0CC CIaL#.J NMj/K-51[1_Qf D-(D-- D<c [U[5(a[U5S:de[U[5(a[U5S:deU(aSSKnURnO[n[US[ 5n[ R"5 /n U"U5Hn UV s/sH"o(aU RU 5OU "U 5PM$ n n U"U 5RU5n URU 5n [R"U SS9RSS9nXR5-nU RU5 M [ R "U SS9RU5n SSS5 U $s sn f!,(df  W $=f)aqBuild zero-shot classifier weights by iterating over class names 1 by 1 Args: model: CLIP model instance tokenizer: CLIP tokenizer instance classnames: A sequence of class (label) names templates: A sequence of callables or format() friendly strings to produce templates per class name device: Device to use. use_tqdm: Enable TQDM progress bar. rNrrr)r6rr"r8r r7r9r:r#r$r%Frr'r(appendstack)r0r2rrrrr8r>r3r@ classnamer-r.r/class_embeddings r!build_zero_shot_classifier_legacyrHGsT" i * *s9~/AA A j( + +J!0CC CII  IaL#.J ":.Igpqgp[c:X__Y/8ICVVgpEqe$''/E$007 kk*:CHHQHOO 335 5O  # #O 4 /!;;'7Q?BB6J  r  sE4%)E/BE4/E44 F) cpuF)rJF) functoolsr itertoolsrtypingrrrrr r9torch.nn.functionalnn functionalrCrr7intrboolrArHrrUs<<  "02+0/SM/E(C-01 / (} / c5<<'( //n,1 &SM&E(C-01 & c5<<'( &  &rT