Setting

logs/train_example/events.out.tfevents.1717960602.federatedlearning.3722452.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8417d062e0aaba975614fc16065721ed4055f234a927bad8be2ef2e1b439b4da
-size 595328

pipeline.py

@@ -26,14 +26,14 @@ class DDPMConditionalPipeline(DiffusionPipeline):
         if isinstance(self.unet.sample_size, int):
             image_shape = (
                 batch_size,
-                self.unet.in_channels,
+                self.unet.config.in_channels,
                 self.unet.sample_size,
                 self.unet.sample_size,
             )
         else:
             image_shape = (
                 batch_size,
-                self.unet.in_channels,
+                self.unet.config.in_channels,
                 *self.unet.sample_size,
             )