7 Commits

Author	SHA1	Message	Date
JB (Don)	54a2361a29	Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True (#29024) ... * Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True * Testing for the non-safe-tensors case, since the default is safe-tensors already * Running fixup/fix-copies * Adding accelerate annotations to tests	2024-05-07 11:12:21 +02:00
NielsRogge	de8bec6df3	[AutoBackbone] Add test (#26094) ... * Add test * Add config_class	2023-09-18 23:47:54 +02:00
Sylvain Gugger	f295fc8a16	Fix last models for common tests that are too big. (#25058) ... * Fix last models for common tests that are too big. * Remove print statement	2023-07-25 07:56:04 -04:00
Sylvain Gugger	1023705440	Check models used for common tests are small (#24824) ... * First models * Conditional DETR * Treat DETR models, skip others * Skip LayoutLMv2 as well * Fix last tests	2023-07-14 14:43:19 -04:00
Sylvain Gugger	89b6ee49fd	Finishing tidying keys to ignore on load (#24535)	2023-06-27 21:35:15 -04:00
Yih-Dar	c23d131eab	Update tiny models for pipeline testing. (#24364) ... * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>	2023-06-20 14:43:10 +02:00
amyeroberts	a717e0318c	Add TimmBackbone model (#22619) ... * Add test_backbone for convnext * Add TimmBackbone model * Add check for backbone type * Tidying up - config checks * Update convnextv2 * Tidy up * Fix indices & clearer comment * Exceptions for config checks * Correclty update config for tests * Safer imports * Safer safer imports * Fix where decorators go * Update import logic and backbone tests * More import fixes * Fixup * Only import all_models if torch available * Fix kwarg updates in from_pretrained & main rebase * Tidy up * Add tests for AutoBackbone * Tidy up * Fix import error * Fix up * Install nattan in doc_test_job * Revert back to setting self._out_xxx directly * Bug fix - out_indices mapping from out_features * Fix tests * Dont accept output_loading_info for Timm models * Set out_xxx and don't remap * Use smaller checkpoint for test * Don't remap timm indices - check out_indices based on stage names * Skip test as it's n/a * Apply suggestions from code review Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * Cleaner imports / spelling is hard --------- Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>	2023-06-06 17:11:30 +01:00