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