Changing init in questionable ways

models/pipeline_loader.py

@@ -18,11 +18,7 @@ def get_pipeline(model_path):
     else:
         # Assume it's a Hugging Face Hub model ID
         # Try to load config to determine if it's text-conditional
-        print(model_path)
-        print("\n\n\n\n")
         config = DiffusionPipeline.load_config(model_path)
-        print(config)
-        print("\n\n\n\n")
         has_text_encoder = "text_encoder" in config
             
         if has_text_encoder:

models/text_diffusion_pipeline.py

@@ -22,18 +22,20 @@ class PipelineOutput(NamedTuple):
 # Create a custom pipeline for text-conditional generation
 class TextConditionalDDPMPipeline(DDPMPipeline):
     def __init__(self, unet, scheduler, text_encoder=None, tokenizer=None, supports_pretrained_split=False, block_embeddings=None):
-        super().__init__(unet=unet, scheduler=scheduler)
+        # Don't call super().__init__() with arguments, call it without arguments first
+        super(DiffusionPipeline, self).__init__()
+        
         self.text_encoder = text_encoder
         self.tokenizer = tokenizer
         self.supports_negative_prompt = hasattr(unet, 'negative_prompt_support') and unet.negative_prompt_support
         self.supports_pretrained_split = supports_pretrained_split
         self.block_embeddings = block_embeddings
-
+        
         if self.tokenizer is None and self.text_encoder is not None:
             # Use the tokenizer from the text encoder if not provided
             self.tokenizer = self.text_encoder.tokenizer
         
-        # Register the text_encoder so that .to(), .cpu(), .cuda(), etc. work correctly
+        # Register ALL modules at once, including the ones from the parent class
         self.register_modules(
             unet=unet,
             scheduler=scheduler,