1812 Commits

Author	SHA1	Message	Date
Ross Wightman	e7925ea6d6	Add explicit disable of fc_norm to pe giant spatial variant	2025-05-21 07:31:36 -07:00
Ross Wightman	f14f6507ab	Merge pull request #2487 from huggingface/eva_pe_integration ... Add EVA ViT based PE (Perceptual Encoder) impl	2025-05-15 12:56:03 -07:00
Ross Wightman	16d0b26e19	Fix torchscript issue with legacy tnt	2025-05-14 09:33:41 -07:00
Ross Wightman	74ad32a67e	Updated tnt model weights on hub, add back legacy model in case bwd compat	2025-05-14 08:40:53 -07:00
Ryan	69b1fbcdc1	Merge branch 'main' into tnt	2025-05-11 22:45:48 +08:00
Ross Wightman	88b7ef6035	Disable dynamic_img_size default on PE models for now	2025-05-10 15:00:29 -07:00
Ross Wightman	fa7287faee	Add EVA ViT based PE (Perceptual Encoder) impl	2025-05-10 13:13:12 -07:00
Ryan Ramos	81900a6bae	remove download argument from torch_kwargs	2025-05-09 07:34:32 -07:00
Ryan	d1140c1a0f	fix pit & add to test	2025-05-08 07:17:19 -07:00
Ryan	2e9b2a76fb	update some model	2025-05-08 07:17:19 -07:00
Ryan	99c25fa5c0	fix nest type error	2025-05-08 07:17:19 -07:00
Ryan	72b2a09106	fix final norm only apply at last indice	2025-05-08 07:17:19 -07:00
Ryan	98d9cde8b4	support efficientformer_v2	2025-05-08 07:17:19 -07:00
Ryan	12def0d118	support efficientvit, edgenext, davit	2025-05-08 07:17:19 -07:00
Ryan	9aedecbb5f	fix norm at last feat_idx	2025-05-08 07:17:19 -07:00
Ryan	e16d385592	support more models ... inception_next, hgnet, gcvit, focalnet, inception_v4	2025-05-08 07:17:19 -07:00
Ryan	6b6beffa6b	support mambaout, metaformer, nest, nextvit, pvt_v2	2025-05-08 07:17:19 -07:00
Ryan	2d3155908c	fix tresnet and rdnet	2025-05-08 07:17:19 -07:00
Ryan	f8be741f0f	support rexnet, resnetv2, repvit and repghostnet	2025-05-08 07:17:19 -07:00
Ryan	5e8cc616d4	support tiny_vit	2025-05-08 07:17:19 -07:00
Ryan	8befebd93c	support vovnet	2025-05-08 07:17:19 -07:00
Ryan	411b892dbc	support tresnet	2025-05-08 07:17:19 -07:00
Ryan	37bbac1256	Fix checkpoint_filter_fn	2025-05-03 15:41:18 +08:00
Ryan	fc0b6ad183	Fix default_cfgs	2025-05-03 08:40:07 +08:00
Ryan	848b8c3e57	Support features_only	2025-05-02 20:59:05 +08:00
Ryan	b37f0f7a76	Update tnt.py	2025-05-02 20:34:31 +08:00
Xihua Dong	0c136f7dab	fix img_size type	2025-04-18 21:05:05 -07:00
Ross Wightman	3ff3899026	Add local-dir: schema support for model loading (config + weights) from folder	2025-04-17 12:32:19 -07:00
Ross Wightman	ceca5efdec	Remove torch_out from onnx export, no point without the export_ fn	2025-04-15 14:01:17 -07:00
Ross Wightman	0cae8a4cd8	Fix #2472, torch.onnx.export_ (with return output) finally removed :(	2025-04-15 14:01:17 -07:00
Ross Wightman	681be882e8	Fix arg merging of sknet, old seresnet. Fix #2470	2025-04-14 10:32:26 -07:00
Ross Wightman	98e9651952	Update version.py ... Version 1.0.15, prep for a release	2025-02-22 10:50:21 -08:00
Adam J. Stewart	92682d8d4d	timm.models: explicitly export attributes	2025-02-21 14:19:39 -08:00
Ross Wightman	a667d3d8f0	siglip2 weights on hub, fix forward_intermediates when no prefix tokens (& return prefix selected)	2025-02-21 13:10:51 -08:00
Ross Wightman	f63a11cf81	Remove duplicate so400m/16 @ 256 model def	2025-02-21 13:10:51 -08:00
Ross Wightman	9758e0b8b0	Prep for siglip2 release	2025-02-21 13:10:51 -08:00
Adam J. Stewart	c68d724e9c	adapt_input_conv: add type hints	2025-02-21 12:28:22 -08:00
Ross Wightman	105a667baa	Dev version 1.0.15.dev0	2025-02-17 15:50:12 -08:00
Ross Wightman	7234f5c6c5	Add 448 so150m2 weight/model, add updated internvit 300m weight	2025-02-17 12:59:10 -08:00
Ross Wightman	9ce824c39a	Add vit so150m2 weights	2025-02-14 15:55:51 -08:00
Ross Wightman	490d222dd8	Fix issue taking device from V before V exists	2025-01-31 12:52:47 -08:00
Lucas Nestler	e025328f96	simplify RNG	2025-01-31 17:26:14 +01:00
Lucas Nestler	6367267298	unify RNG	2025-01-31 17:23:53 +01:00
Ross Wightman	872978ccfe	Fix comment, add 'stochastic weight decay' idea because why not	2025-01-30 18:22:36 -08:00
Ross Wightman	510bbd5389	Change start/end args	2025-01-30 18:22:36 -08:00
Ross Wightman	31831f5948	Change flattening behaviour in Kron	2025-01-30 18:22:36 -08:00
Ross Wightman	b3a83b81d6	Prep Kron for merge, add detail to attributions note, README.	2025-01-27 21:02:26 -08:00
Ross Wightman	67ef6f0a92	Move opt_einsum import back out of class __init__	2025-01-27 21:02:26 -08:00
Ross Wightman	9ab5464e4d	More additions to Kron	2025-01-27 21:02:26 -08:00
Ross Wightman	5f10450235	Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now.	2025-01-27 21:02:26 -08:00