diffusers
Fix multi-gpu case for `train_cm_ct_unconditional.py` in terms of unwrapping `unet` model
#8653
Merged

Fix multi-gpu case for `train_cm_ct_unconditional.py` in terms of unwrapping `unet` model #8653

tolgacangoz
tolgacangoz Fix multi-gpu case
df1ad869
tolgacangoz tolgacangoz marked this pull request as draft 1 year ago
tolgacangoz Merge branch 'main' into fix-multi-gpu-ct
6aab9bec
tolgacangoz Merge branch 'main' into fix-multi-gpu-ct
6c0cca61
tolgacangoz tolgacangoz marked this pull request as ready for review 1 year ago
tolgacangoz Merge branch 'main' into fix-multi-gpu-ct
fd5a80ba
tolgacangoz Merge branch 'main' into fix-multi-gpu-ct
604a7082
a-r-r-o-w a-r-r-o-w requested a review from sayakpaul sayakpaul 1 year ago
sayakpaul
sayakpaul approved these changes on 2024-07-14
HuggingFaceDocBuilderDev
tolgacangoz Prefer previously created `unwrap_model()` function
4b06b954
tolgacangoz Merge branch 'main' into fix-multi-gpu-ct
4d8fcec2
tolgacangoz
a-r-r-o-w
a-r-r-o-w approved these changes on 2024-07-17
tolgacangoz `chore: update unwrap_model() function to use accelerator.unwrap_mode…
62f1ff3a
tolgacangoz tolgacangoz requested a review from a-r-r-o-w a-r-r-o-w 1 year ago
tolgacangoz
a-r-r-o-w
a-r-r-o-w a-r-r-o-w merged c1dc2ae6 into main 1 year ago
tolgacangoz
tolgacangoz tolgacangoz deleted the fix-multi-gpu-ct branch 1 year ago
tolgacangoz tolgacangoz changed the title Fix multi-gpu case for `train_cm_ct_unconditional.py` Fix multi-gpu case for `train_cm_ct_unconditional.py` in terms of unwrapping `unet` model 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone