Update pipeline.py

pipeline.py

@@ -10,7 +10,7 @@ from diffusers.configuration_utils import ConfigMixin
 class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
     """SuperDiffPipeline."""
 
-    def __init__(self, unet: Callable, vae: Callable, text_encoder: Callable, scheduler: Callable, tokenizer: Callable, **kwargs) -> None:
+    def __init__(self, unet: Callable, vae: Callable, text_encoder: Callable, scheduler: Callable, tokenizer: Callable) -> None:
         """__init__.
 
         Parameters
@@ -33,7 +33,6 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
         None
 
         """
-        print("hello this is a test", flush=True)
         super().__init__()
         # Register additional parameters for flexibility
         # Explicitly assign required components
@@ -44,7 +43,6 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
         #self.scheduler = scheduler
 
         device = "cuda" if torch.cuda.is_available() else "cpu"
-        print("hello this is a test", flush=True)
         vae.to(device)
         unet.to(device)
         text_encoder.to(device)
@@ -53,7 +51,6 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
                               vae=vae, 
                               text_encoder=text_encoder,
                               tokenizer=tokenizer,
-                              **kwargs
                               )
 
         #self.register_to_config(