Spaces:

yslan
/

LN3Diff_I23D

Running on Zero

NIRVANALAN commited on Aug 19, 2024

Commit

00d21f3

1 Parent(s): 0492152

update

Files changed (1) hide show

guided_diffusion/train_util.py CHANGED Viewed

@@ -134,7 +134,7 @@ class TrainLoop:
             self.use_ddp = True
             self.ddpm_model = self.model
             self.ddp_model = DDP(
-                self.model,
                 device_ids=[dist_util.dev()],
                 output_device=dist_util.dev(),
                 broadcast_buffers=False,

             self.use_ddp = True
             self.ddpm_model = self.model
             self.ddp_model = DDP(
+                self.model.to(dist_util.dev()),
                 device_ids=[dist_util.dev()],
                 output_device=dist_util.dev(),
                 broadcast_buffers=False,