9 Commits

Author	SHA1	Message	Date
amyeroberts	1de7dc7403	Skip tests properly (#31308) ... * Skip tests properly * [test_all] * Add 'reason' as kwarg for skipTest * [test_all] Fix up * [test_all]	2024-06-26 21:59:08 +01:00
Yih-Dar	a7cb92aa03	fix / skip (for now) some tests before switch to torch 2.2 (#28838) ... * fix / skip some tests before we can switch to torch 2.2 * style --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>	2024-02-02 14:11:50 +01:00
Angela Yi	243e186efb	Add serialization logic to pytree types (#27871) ... * Add serialized type name to pytrees * Modify context * add serde test	2024-01-29 10:41:20 +01:00
Xuehai Pan	cc7803c0a6	Register ModelOutput as supported torch pytree nodes (#26618) ... * Register ModelOutput as supported torch pytree nodes * Test ModelOutput as supported torch pytree nodes * Update type hints for pytree unflatten functions	2023-10-24 11:02:40 +02:00
Craig Chan	d7bd325b5a	Add missing Maskformer dataclass decorator, add dataclass check in ModelOutput for subclasses (#25638) ... * Add @dataclass to MaskFormerPixelDecoderOutput * Add dataclass check if subclass of ModelOutout * Use unittest assertRaises rather than pytest per contribution doc * Update src/transformers/utils/generic.py per suggested change Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>	2023-09-14 10:30:49 +01:00
Matthew Hoffman	d4bd33cc9f	Register ModelOutput subclasses as supported torch.utils._pytree nodes (#25358) ... * Register ModelOutput subclasses as supported torch.utils._pytree nodes Fixes #25357 where DDP with static_graph=True does not sync gradients when calling backward() over tensors contained in ModelOutput subclasses * Add test for torch pytree ModelOutput serialization and deserialization	2023-08-08 08:12:11 +02:00
Sylvain Gugger	afce73bd9d	Fix ModelOutput instantiation when there is only one tuple (#20416)	2022-11-23 15:09:21 -05:00
Sylvain Gugger	4975002df5	Reorganize file utils (#16264) ... * Split file_utils in several submodules * Fixes * Add back more objects * More fixes * Who exactly decided to import that from there? * Second suggestion to code with code review * Revert wront move * Fix imports * Adapt all imports * Adapt all imports everywhere * Revert this import, will fix in a separate commit	2022-03-23 10:26:33 -04:00
Lysandre Debut	29c10a41d0	[Test refactor 1/5] Per-folder tests reorganization (#15725) ... * Per-folder tests reorganization Co-authored-by: sgugger <sylvain.gugger@gmail.com> Co-authored-by: Stas Bekman <stas@stason.org>	2022-02-23 15:46:28 -05:00