Make dynamo wrapped modules work with save_pretrained - #2726

Merged

pcuenca merged 12 commits into

mainfrom

dynamo-save-pretrained

Mar 28, 2023

Commits on Mar 17, 2023

Workaround for saving dynamo-wrapped models.
pcuenca
committed
72f1b13

Commits on Mar 20, 2023

Accept suggestion from code review

pcuenca
and
patrickvonplaten
authored
9351bc8
Apply workaround when overriding pipeline components.
pcuenca
committed
351b47b
Ensure the correct config.json is saved to disk.
pcuenca
committed
cb76bde
Save correct module (not compiled one)
pcuenca
committed
e990b53
Add test
pcuenca
committed
f803679
style
pcuenca
committed
ed7a625
fix docstrings
pcuenca
committed
0f3a8f0
Go back to using string comparisons.
pcuenca
committed
2c4ba90

Commits on Mar 21, 2023

Simple test for save_pretrained of compiled models.
pcuenca
committed
4680faf

Commits on Mar 23, 2023

Merge branch 'main' into dynamo-save-pretrained
patrickvonplaten
authored
f00d4d5

Commits on Mar 25, 2023

Helper function to test whether module is compiled.
pcuenca
committed
e41331b