Update syngen_diffusion_pipeline.py

syngen_diffusion_pipeline.py

@@ -104,7 +104,7 @@ class SynGenDiffusionPipeline(StableDiffusionPipeline):
             callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
             callback_steps: int = 1,
             cross_attention_kwargs: Optional[Dict[str, Any]] = None,
-            syngen_step_size: int = 20,
+            syngen_step_size: float = 20.0,
     ):
         r"""
         Function invoked when calling the pipeline for generation.
@@ -492,4 +492,4 @@ def _get_attention_maps_list(
         attention_maps[:, :, i] for i in range(attention_maps.shape[2])
     ]
 
-    return attention_maps_list
+    return attention_maps_list