Skip to content

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

Merged
a-r-r-o-w merged 8 commits into
huggingface:mainfrom
tolgacangoz:fix-multi-gpu-ct
Jul 17, 2024
Merged

Fix multi-gpu case for train_cm_ct_unconditional.py in terms of unwrapping unet model#8653
a-r-r-o-w merged 8 commits into
huggingface:mainfrom
tolgacangoz:fix-multi-gpu-ct

`chore: update unwrap_model() function to use accelerator.unwrap_mode…

62f1ff3
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs