Update pipeline.py

pipeline.py

@@ -46,11 +46,11 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
         self.text_encoder.to(device)
 
         self.register_to_config(
-            #model=model,
-            #vae=vae,
-            #tokenizer=tokenizer,
-            #text_encoder=text_encoder,
-            #scheduler=scheduler,
+            vae=vae.__class__.__name__,
+            scheduler=scheduler.__class__.__name__,
+            tokenizer=tokenizer.__class__.__name__,
+            unet=unet.__class__.__name__,
+            text_encoder=text_encoder.__class__.__name__
             device=device,
             batch_size=None,
             num_inference_steps=None,