diffusers
Make dynamo wrapped modules work with save_pretrained
#2726

Merged

Make dynamo wrapped modules work with save_pretrained #2726

pcuenca merged 12 commits into main from dynamo-save-pretrained

Workaround for saving dynamo-wrapped models.

72f1b13e

pcuenca requested a review from

patrickvonplaten 2 years ago

patrickvonplaten commented on 2023-03-18

Accept suggestion from code review

9351bc82

Apply workaround when overriding pipeline components.

351b47b5

Ensure the correct config.json is saved to disk.

cb76bdee

Save correct module (not compiled one)

e990b53d

Add test

f803679e

style

ed7a625a

pcuenca requested a review from

patrickvonplaten 2 years ago

pcuenca requested a review from

sayakpaul 2 years ago

fix docstrings

0f3a8f0d

Go back to using string comparisons.

2c4ba90a

sayakpaul approved these changes on 2023-03-21

patrickvonplaten commented on 2023-03-21

Simple test for save_pretrained of compiled models.

4680fafa

Merge branch 'main' into dynamo-save-pretrained

f00d4d56

patrickvonplaten approved these changes on 2023-03-23

Helper function to test whether module is compiled.

e41331b7

pcuenca merged 81125d84 into main 2 years ago

pcuenca deleted the dynamo-save-pretrained branch 2 years ago

Reviewers

sayakpaul

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

diffusers Make dynamo wrapped modules work with save_pretrained #2726 Merged

Make dynamo wrapped modules work with save_pretrained #2726

diffusers
Make dynamo wrapped modules work with save_pretrained
#2726

Merged