Update pipeline.py

pipeline.py

@@ -6,26 +6,30 @@ import torch
 
 class VidToMePipeline(DiffusionPipeline):
     # def __init__(self, device="cuda", sd_version="2.1", float_precision="fp16", height=512, width=512):
-    #     # this will initlize the core pipeline components
-    #     pipe, scheduler, model_key = init_model(device, sd_version, None, "none", float_precision)
-    #     self.pipe = pipe
-    #     self.scheduler = scheduler
-    #     self.model_key = model_key
-    #     self.device = device
-    #     self.sd_version = sd_version
-    #     self.float_precision = float_precision
-    #     self.height = height
-    #     self.width = width
+        # # this will initlize the core pipeline components
+        # pipe, scheduler, model_key = init_model(device, sd_version, None, "none", float_precision)
+        # self.pipe = pipe
+        # self.scheduler = scheduler
+        # self.model_key = model_key
+        # self.device = device
+        # self.sd_version = sd_version
+        # self.float_precision = float_precision
+        # self.height = height
+        # self.width = width
 
     def __init__(self, device="cuda", sd_version="1.5", float_precision="fp16", height=512, width=512):
         # Register configuration parameters
         self.register_to_config(device=device, sd_version=sd_version, float_precision=float_precision, height=height, width=width)
-        
-        # Now you can safely use self.device and other attributes
         self.sd_version = sd_version
         self.float_precision = float_precision
         self.height = height
         self.width = width
+        # this will initlize the core pipeline components
+        pipe, scheduler, model_key = init_model(device, sd_version, None, "none", float_precision)
+        self.pipe = pipe
+        self.scheduler = scheduler
+        self.model_key = model_key
+        self.device = device
 
     def __call__(self, video_path=None, video_prompt=None, edit_prompt=None, 
                  control_type="none", n_timesteps=50, guidance_scale=7.5,