Super-squash branch 'main' using huggingface_hub

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,39 @@
+---
+license: apache-2.0
+library_name: timm
+---
+# IdolSankaku EVA02-Large Tagger v1
+
+Supports ratings, characters and general tags.
+
+Trained using https://github.com/SmilingWolf/JAX-CV.  
+TPUs used for training kindly provided by the [TRC program](https://sites.research.google/trc/about/).
+
+## Dataset
+Trained on a human annotated dataset of real world photos.
+
+## Validation results
+`v1.0: P=R: threshold = 0.4985, F1 = 0.6017`
+
+## What's new
+Model v1.0/Dataset v1:  
+First version of the dataset, tags updated on 2024-08-31.  
+`timm` compatible! Load it up and give it a spin using the canonical one-liner!  
+ONNX model is compatible with code developed for the v3 series of WD tagger models.  
+The batch dimension of the ONNX model is not fixed to 1 anymore. Now you can go crazy with batch inference.  
+Switched to Macro-F1 to measure model performance since it gives me a better gauge of overall training progress.
+
+# Runtime deps
+ONNX model requires `onnxruntime >= 1.17.0`
+
+# Inference code examples
+For timm: https://github.com/neggles/wdv3-timm  
+For ONNX: https://huggingface.co/spaces/SmilingWolf/wd-tagger  
+For JAX: https://github.com/SmilingWolf/wdv3-jax
+
+## Final words
+Subject to change and updates.  
+Downstream users are encouraged to use tagged releases rather than relying on the head of the repo.
+
+## Thanks
+Thanks to the whole DeepGHS team for data gathering and encouraging me to push the models much further than they had any reason to attempt to reach, much less succeed.

config.json

@@ -0,0 +1,44 @@
+{
+  "architecture": "eva02_large_patch14_448",
+  "num_classes": 573,
+  "num_features": 1024,
+  "global_pool": "avg",
+  "model_args": {
+    "img_size": 480,
+    "ref_feat_shape": [
+      48,
+      48
+    ],
+    "patch_size": 10,
+    "class_token": false,
+    "num_reg_tokens": 4,
+    "use_abs_pos_emb": false,
+    "scale_mlp": false
+  },
+  "pretrained_cfg": {
+    "custom_load": false,
+    "input_size": [
+      3,
+      480,
+      480
+    ],
+    "fixed_input_size": false,
+    "interpolation": "bicubic",
+    "crop_pct": 1.0,
+    "crop_mode": "center",
+    "mean": [
+      0.5,
+      0.5,
+      0.5
+    ],
+    "std": [
+      0.5,
+      0.5,
+      0.5
+    ],
+    "num_classes": 573,
+    "pool_size": null,
+    "first_conv": null,
+    "classifier": null
+  }
+}

model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:230f2c8365a962d3fa477a971d6f78fe9241339a83c29d8822c236c25411fc4c
+size 1213767025

model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf829d030ef498ba7215ce8e0795d53f69cade917ef2e48f19a8a37f071ca5c6
+size 1213133469

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:73f67eb5b57e8c5b447babc35dd4171993f76724164345385d12b944af28ff4f
+size 1212715828

selected_tags.csv

@@ -0,0 +1,574 @@
+tag_id,name,category,count
+9999998,safe,9,651810
+9999997,questionable,9,68432
+9999996,explicit,9,198664
+12109,1girl,0,238218
+93,swimsuit,0,237483
+12204,asian,0,194878
+26,cleavage,0,184922
+237,skirt,0,175256
+6995,solo,0,169756
+10217,female,0,154448
+3,bikini,0,148660
+13,thighhighs,0,129107
+420,blouse,0,124602
+32352,female_only,0,124246
+62,dress,0,108826
+9,pantyhose,0,98410
+6501,oshiri,0,97207
+6538,breasts,0,90868
+10,school_uniform,0,83387
+35,twintails,0,81407
+353,pleated_skirt,0,81163
+6502,pantsu,0,73753
+268,bed,0,73712
+235,miniskirt,0,72734
+42,blonde_hair,0,71862
+6593,long_hair,0,65183
+298,brown_hair,0,63463
+409,side-tie_bikini,0,60527
+44642,east_asian,0,58319
+16,zettai_ryouiki,0,56237
+84,shorts,0,50824
+11243,non-asian,0,50232
+19,hairband,0,49120
+25,boots,0,49089
+28,gloves,0,47776
+106,bikini_top,0,46413
+34381,female_solo,0,46028
+1882,nude,0,45068
+15705,japanese,0,43500
+6500,nipples,0,42376
+6511,large_breasts,0,41494
+11154,looking_at_viewer,0,39065
+55,bra,0,38878
+6536,black_hair,0,38016
+740,ponytail,0,37725
+6655,hair_bow,0,36304
+6521,vagina,0,33724
+22,sailor_uniform,0,33277
+417,open_clothes,0,32976
+186,kneesocks,0,32936
+630,one-piece_swimsuit,0,32884
+683,wet,0,32869
+51,apron,0,31398
+26980,caucasian,0,30824
+6541,brown_eyes,0,29492
+398,choker,0,29176
+36,animal_ears,0,29150
+24645,sheer_legwear,0,28849
+626,detached_sleeves,0,28114
+8699,megane,0,26055
+16991,black_legwear,0,25852
+916,bathroom,0,24966
+11270,crop_top,0,24517
+6786,navel,0,24192
+170,pink_hair,0,23904
+698,fishnets,0,22786
+923,barefoot,0,22544
+2551,shirt,0,22412
+248,feet,0,22189
+27,elbow_gloves,0,22066
+4528,smile,0,22041
+493,tie,0,21791
+309,red_hair,0,21680
+727,vest,0,21449
+182,blue_hair,0,20617
+6563,thighs,0,20570
+6507,sitting,0,20241
+160,maid,0,19960
+1505,stockings,0,19709
+462,fingerless_gloves,0,19381
+161,maid_uniform,0,19140
+6543,spread_legs,0,19018
+204,tshirt,0,18495
+111,hat,0,18006
+22974,non-nude,0,17902
+849,bodysuit,0,17795
+119,camisole,0,17226
+52,socks,0,17124
+6583,small_breasts,0,16805
+6522,anus,0,16498
+63,purple_hair,0,16392
+20,hair_ribbons,0,16387
+6899,blue_eyes,0,16318
+1137,bikini_bottom,0,16311
+144,blazer,0,16225
+6533,short_hair,0,16129
+267,aqua_hair,0,15837
+16407,indoors,0,15833
+75,garter_belt,0,15384
+625,tank_top,0,15347
+2114,legs,0,15205
+6592,dark_skin,0,15128
+513,see-through,0,15118
+345,fishnet_stockings,0,14692
+16596,clothing,0,14569
+13135,white_legwear,0,14464
+76,lingerie,0,14368
+202,jacket,0,13535
+273,cat_ears,0,13518
+3461,lying,0,13466
+5611,high_heels,0,13402
+6577,medium_breasts,0,13262
+1580,bathtub,0,13078
+6506,pubic_hair,0,12956
+1028,necklace,0,12603
+74,sweater,0,12469
+34383,shaved_vagina,0,12463
+368,skirt_lift,0,12378
+148,twin_braids,0,12365
+681,beach,0,12175
+922,teeth,0,12138
+873,tattoo,0,12029
+176,silver_hair,0,11781
+155,collar,0,11774
+7147,armpit,0,11766
+138,bow,0,11340
+6508,huge_breasts,0,11295
+82,gym_uniform,0,11211
+9386,toes,0,11172
+128,underboob,0,10739
+304,wafuku,0,10487
+1407,tail,0,10381
+3779,jumper,0,10238
+3201,nail_polish,0,10235
+197,cardigan,0,10218
+6678,black_eyes,0,10127
+6532,2girls,0,10085
+2371,topless,0,9799
+10165,brunette,0,9594
+1524,halter_top,0,9589
+1390,denim,0,9465
+373,braid,0,9404
+6598,open_mouth,0,9326
+37,bunny_ears,0,9253
+80,buruma,0,9152
+92,scarf,0,9072
+6337,kneeling,0,8874
+365,side_ponytail,0,8874
+6783,back,0,8868
+2055,lipstick,0,8838
+5521,thigh_boots,0,8816
+297,orange_hair,0,8756
+1968,monokini,0,8734
+1548,ocean,0,8621
+1843,leotard,0,8588
+212,ahoge,0,8582
+6560,erect_nipples,0,8563
+1326,lips,0,8532
+18408,curvaceous,0,8517
+136,striped,0,8367
+21,kneehighs,0,8284
+17084,large_ass,0,8219
+125,shirt_lift,0,8206
+17011,thick_thighs,0,8194
+6784,looking_back,0,8163
+7863,soles,0,8116
+7299,underwear,0,7953
+153,bottomless,0,7951
+189,green_hair,0,7929
+1225,wings,0,7919
+6910,big_ass,0,7901
+6548,bowtie,0,7837
+303,corset,0,7804
+2120,makeup,0,7796
+24766,medium_hair,0,7756
+3462,standing,0,7748
+6926,piercing,0,7738
+696,mirror,0,7709
+854,leggings,0,7658
+742,sideboob,0,7655
+1481,ribbon,0,7653
+6166,bracelet,0,7651
+247,close-up,0,7612
+6542,eyes_closed,0,7564
+123,midriff,0,7493
+511,headphones,0,7489
+2723,headset,0,7461
+30,nurse,0,7452
+453,kimono,0,7344
+6589,on_back,0,7240
+415,tubetop,0,7126
+33,nurse_uniform,0,7124
+1857,no_bra,0,7049
+1287,white_hair,0,7033
+135,pool,0,6944
+6564,from_behind,0,6897
+317,thong,0,6875
+6700,shoes,0,6847
+39,fishnet_pantyhose,0,6761
+17,paipan,0,6721
+31,nurse_cap,0,6635
+1147,couch,0,6631
+18284,teasing,0,6597
+9549,multiple_girls,0,6505
+8036,outdoors,0,6379
+2665,multi-colored_hair,0,6220
+34382,completely_nude,0,6172
+715,beret,0,6073
+1766,bookbag,0,5921
+12499,bangs,0,5913
+50690,caucasion,0,5745
+78,hoodie,0,5716
+751,flowers,0,5655
+100,waitress,0,5597
+6950,side,0,5386
+101,waitress_uniform,0,5384
+523,cape,0,5379
+6567,chinese,0,5367
+456,crossplay,0,5350
+10094,pale_skin,0,5332
+25216,nude_female,0,5251
+68,uniform,0,5219
+36547,light-skinned_female,0,5212
+1489,sleeveless_sweater,0,5130
+9049,areolae,0,5068
+6735,spread_pussy,0,5029
+6669,bent_over,0,5008
+6683,arms_up,0,5007
+307,sword,0,4984
+274,garter,0,4921
+570,qipao,0,4834
+13222,curves,0,4614
+334,gun,0,4556
+53,babydoll,0,4545
+216,pants,0,4521
+8563,ribbon_tie,0,4438
+2583,sleeveless_blouse,0,4373
+2338,undressing,0,4356
+9078,labia,0,4336
+925,water,0,4327
+11886,on_bed,0,4315
+12723,garter_straps,0,4315
+1640,bound,0,4310
+256,green_eyes,0,4307
+6,armband,0,4256
+1388,dress_lift,0,4236
+1364,microphone,0,4155
+1380,chemise,0,4155
+9575,bedroom,0,4110
+691,coat,0,4026
+15061,non_nude,0,4007
+26125,latina,0,3973
+624,suspenders,0,3968
+1540,chair,0,3925
+6534,nopan,0,3910
+13249,curvy,0,3909
+6769,all_fours,0,3894
+169,bare_shoulders,0,3867
+7613,gauntlets,0,3835
+8810,black_thighhighs,0,3815
+2461,rope,0,3789
+1124,horns,0,3778
+1406,stuffed_animal,0,3720
+8942,skinny,0,3715
+2161,top_hat,0,3677
+461,belt,0,3669
+701,shawl,0,3662
+7431,arm_up,0,3583
+312,red_eyes,0,3546
+60,cuffs,0,3498
+6690,on_side,0,3494
+6616,hair_bun,0,3478
+126,torn_clothes,0,3461
+1282,hair_clips,0,3349
+24648,striped_legwear,0,3308
+6927,navel_piercing,0,3306
+1108,pillow,0,3262
+262,loose_socks,0,3259
+6803,wig,0,3236
+6605,mole,0,3231
+12354,pink_eyes,0,3208
+205,turtleneck,0,3175
+301,kunoichi,0,3174
+166,chains,0,3157
+1041,school_swimsuit,0,3154
+8046,tanned,0,3131
+1122,side-tie_panties,0,3116
+12962,huge_ass,0,3114
+6514,bondage,0,3101
+475,mask,0,3081
+1244,headband,0,3036
+281,jeans,0,3030
+184,eyepatch,0,3030
+1445,robe,0,3015
+157,dress_shirt,0,3012
+14310,trousers,0,2981
+11233,plump,0,2945
+1440,microskirt,0,2913
+689,santa_costume,0,2912
+6581,white_panties,0,2889
+34177,teenage,0,2821
+2191,petticoat,0,2806
+2122,earring,0,2801
+8192,white_thighhighs,0,2780
+15239,tiered_skirt,0,2779
+181,yukata,0,2771
+137,sunglasses,0,2734
+6944,weapon,0,2695
+12331,eyeshadow,0,2687
+28931,ebony,0,2651
+12016,wide_hips,0,2648
+6571,open_shirt,0,2611
+25578,cropped_jacket,0,2607
+6701,face,0,2589
+6723,jewelry,0,2557
+10581,long_nails,0,2556
+32864,ass_focus,0,2548
+200,umbrella,0,2526
+3808,legs_up,0,2524
+589,elf_ears,0,2516
+35084,dark-skinned_female,0,2491
+83,looking_over_glasses,0,2472
+32919,female_pubic_hair,0,2461
+669,wristbands,0,2457
+6696,earrings,0,2456
+416,stethoscope,0,2446
+1884,panty_pull,0,2434
+6602,on_all_fours,0,2412
+8866,black_panties,0,2402
+2292,bows,0,2384
+6656,curly_hair,0,2382
+34564,hairy_pussy,0,2376
+6680,squat,0,2375
+1566,polo,0,2336
+6664,on_stomach,0,2304
+13749,petite,0,2291
+735,short_shorts,0,2267
+15373,labia_majora,0,2265
+2632,pasties,0,2257
+2813,hair_pods,0,2251
+9984,muscle,0,2240
+6471,double_bun,0,2226
+1859,veil,0,2217
+2328,shower,0,2186
+120,lace,0,2157
+6572,hair_ornament,0,2139
+28603,tied_hair,0,2133
+2464,oiled,0,2106
+1375,military_uniform,0,2099
+6510,voluptuous,0,2093
+11278,purple_eyes,0,2072
+2569,wristband,0,2069
+1491,plaid,0,2065
+12397,paw_gloves,0,2047
+886,witch_hat,0,2021
+1208,armor,0,1984
+752,handcuffs,0,1976
+7318,ring,0,1966
+7718,tagme_song,0,1956
+242,sandals,0,1926
+670,fan,0,1924
+13075,presenting,0,1922
+6509,looking_down,0,1916
+6737,clitoris,0,1910
+15164,pinup,0,1896
+6698,covering_breasts,0,1888
+2863,polka_dot,0,1878
+697,wet_hair,0,1878
+25540,red_legwear,0,1875
+741,sleeveless,0,1873
+65,ruffles,0,1870
+522,bustier,0,1861
+1964,headdress,0,1838
+6770,happy,0,1828
+308,katana,0,1820
+31916,innie_pussy,0,1811
+6704,tease,0,1796
+865,hairpods,0,1794
+9104,hair_ties,0,1775
+8637,plushie,0,1772
+7201,hand_on_hip,0,1758
+14037,white_shirt,0,1752
+18362,nipple_piercing,0,1741
+2115,tree,0,1739
+5325,upskirt,0,1726
+1921,default_costume,0,1726
+2943,soap_suds,0,1725
+2876,panties_down,0,1696
+7203,wet_clothes,0,1695
+11031,busty,0,1691
+2144,window,0,1686
+31647,muscular_female,0,1683
+31611,female_focus,0,1682
+1387,tiara,0,1677
+1266,lollipop,0,1644
+885,wand,0,1640
+17281,eyeliner,0,1631
+10248,armbands,0,1609
+747,towel,0,1602
+6888,reflection,0,1601
+7095,yellow_eyes,0,1596
+6574,plant,0,1592
+30030,asymmetrical_clothes,0,1583
+12,sky,0,1578
+24742,asymmetrical_legwear,0,1577
+40407,toenail_polish,0,1572
+3788,yaya,0,1567
+2229,latex,0,1565
+8864,single_thighhigh,0,1560
+15201,crinoline,0,1556
+555,striped_panties,0,1550
+734,chopsticks,0,1545
+28148,pov_feet,0,1537
+12501,parted_lips,0,1527
+1334,denim_shorts,0,1507
+6573,butt_crack,0,1498
+24729,fake_breasts,0,1495
+129,bandages,0,1494
+6785,against_wall,0,1494
+1185,hair_ribbon,0,1482
+4612,sling_bikini,0,1477
+16995,breasts_outside,0,1463
+54,bloomers,0,1461
+30459,arm_tattoo,0,1458
+29574,smartphone,0,1452
+30909,foot_fetish,0,1433
+706,cheerleader_uniform,0,1433
+875,leash,0,1428
+812,police_uniform,0,1423
+3240,stocking_cap,0,1420
+1568,backpack,0,1416
+654,chain,0,1410
+233,teddy_bear,0,1407
+6689,leg_up,0,1393
+6525,breast_press,0,1392
+813,policewoman,0,1391
+945,crown,0,1389
+1642,bound_hands,0,1388
+22273,caramel_eyes,0,1387
+26131,partially_visible_vulva,0,1386
+12714,legwear,0,1374
+156,dog_ears,0,1373
+7099,micro_bikini,0,1367
+165,knife,0,1366
+1824,half-skirt,0,1360
+2543,hair_buns,0,1359
+32315,labia_minora,0,1358
+40,guitar,0,1350
+23578,sailor_dress,0,1343
+38,bunny_girl,0,1341
+1730,body_paint,0,1339
+7411,urine,0,1324
+6682,looking_up,0,1300
+6200,uri,0,1293
+6790,wrist_cuffs,0,1290
+1880,long_sleeves,0,1289
+6738,urethra,0,1268
+12226,leaning_forward,0,1265
+19176,head_out_of_frame,0,1264
+24938,blue_legwear,0,1262
+3717,facepaint,0,1257
+948,cellphone,0,1256
+2577,vaulting_horse,0,1253
+6587,maid_headdress,0,1242
+1359,sleeveless_shirt,0,1238
+9449,hand_on_head,0,1227
+1224,goggles,0,1227
+6751,abs,0,1222
+2286,hood,0,1215
+6994,headwings,0,1213
+1989,classroom,0,1212
+6591,tan_skin,0,1207
+6211,sweater_vest,0,1201
+2210,table,0,1188
+6798,tan_lines,0,1186
+8859,aqua_eyes,0,1182
+1146,cowboy_hat,0,1175
+7113,squatting,0,1160
+16595,wooden_floor,0,1156
+15226,scarf_tie,0,1146
+8553,half_skirt_open_front,0,1130
+7104,very_long_hair,0,1123
+39732,foot_tease,0,1121
+29139,solo_focus,0,1118
+707,pom_poms,0,1117
+6520,shiny,0,1112
+2760,hair_clip,0,1106
+8050,ribs,0,1103
+1004,leather,0,1102
+10541,arm_support,0,1100
+16153,fetish,0,1090
+1492,slip,0,1090
+17014,upper_body,0,1078
+14904,grey_eyes,0,1069
+26047,dark_nipples,0,1061
+2005,bath,0,1049
+9367,police_hat,0,1038
+31755,leg_tattoo,0,1030
+6627,stairs,0,1028
+7186,nose_piercing,0,1025
+1630,drill_hair,0,1025
+6841,leg_lift,0,1022
+13123,grin,0,1008
+2013,v,0,999
+35956,blue_swimsuit,0,998
+336,rifle,0,987
+8054,arms_behind_back,0,980
+17037,eyelashes,0,967
+6513,shibari,0,966
+6615,crossed_legs,0,963
+663,bat_wings,0,953
+598,fur,0,946
+28121,straight_hair,0,945
+1486,bell,0,943
+43222,completely_nude_female,0,931
+9016,costume,0,928
+25582,pink_legwear,0,925
+11302,nail_art,0,924
+15595,leaning_back,0,912
+9507,hair_over_one_eye,0,910
+6544,bound_legs,0,902
+25778,blunt_bangs,0,894
+17876,footwear,0,883
+4325,forest,0,881
+994,hair_flower,0,874
+7187,ganguro,0,863
+7031,bound_arms,0,856
+2471,freckles,0,837
+682,sand,0,834
+3380,cat_paws,0,832
+7083,cleavage_cutout,0,830
+6979,covering_pussy,0,821
+19188,pointed_ears,0,795
+1768,zipper,0,787
+6633,people,0,751
+1933,open_jacket,0,733
+7030,arms_behind,0,707
+14431,lip_piercing,0,706
+7402,grass,0,694
+1922,hatsune_miku,4,14200
+3850,sheryl_nome,4,5065
+9885,super_soniko,4,4673
+46,soryu_asuka_langley,4,4642
+23,suzumiya_haruhi,4,3552
+3208,izayoi_sakuya,4,3037
+3414,ranka_lee,4,2831
+2724,kagamine_rin,4,2731
+9916,kaname_madoka,4,2458
+478,saber,4,2424
+4678,megurine_luka,4,2338
+3215,hakurei_reimu,4,2279
+292,ayanami_rei,4,2127
+21763,minami_kotori,4,2120
+19003,yazawa_niko,4,1903
+6925,yoko_littner,4,1849
+24222,shimakaze_(kantai_collection),4,1815
+20023,nishikino_maki,4,1783
+21354,sonoda_umi,4,1755
+5383,makinami_mari_illustrious,4,1587
+9983,akemi_homura,4,1541
+300,kasumi_(doa),4,1527
+5246,nakano_azusa,4,1500
+6971,remilia_scarlet,4,1431
+9588,gokou_ruri,4,1395
+1057,kanu_unchou,4,1389
+35267,yorha_no.2_type_b,4,1275
+9688,tomoe_mami,4,1120
+9243,stocking_(psg),4,1092
+8561,fate_testarossa,4,1068
+843,morrigan_aensland,4,938
+1059,ryomou_shimei,4,887
+22239,kousaka_honoka,4,810
+20361,takanashi_rikka,4,778

sw_jax_cv_config.json

@@ -0,0 +1,22 @@
+{
+    "image_size": 480,
+    "model_name": "eva02_large",
+    "model_args": {
+        "image_size": 480,
+        "patch_size": 10,
+        "num_classes": 573,
+        "use_pos_emb": false,
+        "use_cls_token": false,
+        "num_register_tokens": 4,
+        "num_layers": 24,
+        "embed_dim": 1024,
+        "mlp_dim": 2730,
+        "num_heads": 16,
+        "scale_mlp": false,
+        "drop_path_rate": 0.1,
+        "use_norm_bias": false,
+        "use_linear_bias": true,
+        "norm_layer": "reparam_layernorm",
+        "layer_norm_eps": 1e-06
+    }
+}