DreamLite

Paused

carlofkl commited on Apr 28

Commit

745c3e3

1 Parent(s): 7133397

fix bucket

Files changed (1) hide show

dreamlite/pipelines/dreamlite/pipeline_dreamlite.py CHANGED Viewed

@@ -295,7 +295,7 @@ class DreamLitePipeline(
         return_dict: bool = True,
         max_sequence_length: int = 200,
         text_pad_embedding: Optional[torch.Tensor] = None,
-        bucket: int = 765,
     ):
         # 1. Init pipeline parameters
         height = height or self.default_sample_size * self.vae_scale_factor

         return_dict: bool = True,
         max_sequence_length: int = 200,
         text_pad_embedding: Optional[torch.Tensor] = None,
+        bucket: int = 0,
     ):
         # 1. Init pipeline parameters
         height = height or self.default_sample_size * self.vae_scale_factor