transformers
Refactor `return_dict` logic to remove complicated if/else paths
#36794
Merged

Refactor `return_dict` logic to remove complicated if/else paths #36794

qubvel
qubvel qubvel force pushed from 39a0c4da to 6d1801f6 361 days ago
qubvel SAM
41df5347
qubvel CLIP
49c09043
qubvel SigLIP
c9d8472b
qubvel GOT-OCR2 (depends on SAM)
7a120dfc
qubvel SigLIP2 (depends on SigLIP)
9cca9578
qubvel qubvel force pushed from 6d1801f6 to 9cca9578 361 days ago
qubvel qubvel requested a review from yonigozlan yonigozlan 361 days ago
qubvel qubvel marked this pull request as ready for review 361 days ago
qubvel trigger tests
6fc6e3dd
HuggingFaceDocBuilderDev
qubvel Fix SAM
9df76835
qubvel Fix missed indexing, use named attributes
6fca100f
qubvel Llama
2a0d6974
qubvel Aria
7248a3b8
qubvel Bamba
b3e42a8c
qubvel Update llama: missed outputs return type
d7b86fdb
qubvel (fixup) Aria
eabb7e44
qubvel DiffLlama
3bba0731
qubvel Emu3
c01e8129
qubvel Gemma
945fa4cb
qubvel Gemma2
3cff2247
qubvel Paligemma
8b32936c
qubvel Fix paligemma
9706d65b
qubvel Gemma3
20da41a6
qubvel GLM
8ceea592
qubvel Helium
6ccf8255
qubvel JetMoe
44fc8a07
qubvel Jamba
387dbbc0
qubvel Mistral
ad9480da
qubvel Mistral
5658e588
qubvel Mixtral
41611600
qubvel Nemotron
04c07ff2
qubvel Olmo
37cf83d1
qubvel Olmo2
9a52c864
qubvel Persimmon
f8855715
qubvel Phi
b8c33d36
qubvel Phi3
2ecad9c4
qubvel PhiMoe
17dc3cab
qubvel Qwen2
2b6115a3
qubvel Qwen2_moe
ba21b6e9
qubvel StableLM
e6a8d17b
qubvel Starcoder2
5aa4eec1
qubvel qubvel added Core: Modeling
qubvel Merge branch 'main' into refactor-return-dict
f0184dd6
qubvel
qubvel qubvel requested a review from ArthurZucker ArthurZucker 360 days ago
LysandreJik
LysandreJik approved these changes on 2025-03-19
ArthurZucker
ArthurZucker approved these changes on 2025-03-19
qubvel Merge branch 'main' into refactor-return-dict
264e316d
qubvel
qubvel Merge branch 'main' into refactor-return-dict
25846adc
qubvel Add return_dict decorator
5c4888a2
qubvel SAM
e0457812
qubvel Update decorator: compile, export, trace - friendly
d953f056
qubvel Llama (decorator)
683f1949
qubvel SAM (decorator)
025d06d4
qubvel Add decorator `can_return_tuple`
6c244b25
qubvel Llama
ee0ce9d2
qubvel Merge branch 'return-dict-decorator' into refactor-return-dict
29a0cbe9
qubvel Update to decorator
e1920b8a
qubvel Update CLIP
1dbabf2a
qubvel Update decorator to store `_is_top_level_module` in self
34f3c784
qubvel Update decorator to correctly handle compile/export
7ea77812
qubvel Remove is_torchdynamo_compiling constraint, all work fine with self a…
f0f2987a
qubvel Typing
cdc4e538
qubvel GPT NeoX
c7b02919
qubvel Fixup
256ea025
qubvel Merge branch 'main' into refactor-return-dict
cddd0cde
qubvel Fix attribute Granite
8f6e3e9f
qubvel Fix return type mixtral
c761845f
qubvel Update Gemma3
53599bd2
qubvel Merge branch 'main' into refactor-return-dict
6975f740
qubvel Fix Cohere amd Cohere2
fe4a93a7
qubvel Fixup
18d5b4e4
qubvel qubvel requested a review from ArthurZucker ArthurZucker 355 days ago
qubvel
ArthurZucker
ArthurZucker approved these changes on 2025-03-27
qubvel Merge branch 'main' into refactor-return-dict
5ee0f55a
qubvel Merge branch 'main' into refactor-return-dict
73946f02
qubvel Fix corner case for Phi4, when activation is shared
a653a6ea
qubvel (fix-copies) deepseekv3, phi4
23a5114f
qubvel Fixup
5ce53797
qubvel Merge branch 'main' into refactor-return-dict
e08cfede
qubvel Apply to qwen3/qwen3_moe
630d6b8a
qubvel Merge branch 'main' into refactor-return-dict
09c81d6f
qubvel Fix
62cdb986
qubvel qubvel merged a1e389e6 into main 348 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone