Upload 13 files

pipeline.py

@@ -43,6 +43,7 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
         #self.scheduler = scheduler
 
         device = "cuda" if torch.cuda.is_available() else "cpu"
+
         vae.to(device)
         unet.to(device)
         text_encoder.to(device)
@@ -51,8 +52,8 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
                               vae=vae, 
                               text_encoder=text_encoder,
                               tokenizer=tokenizer,
-                              **kwargs
-                              )
+                              **kwargs)
+
 
         #self.register_to_config(
         #    vae=vae.__class__.__name__,