jadechoghari
/

VidToMe

Model card Files Files and versions Community

jadechoghari commited on Oct 7

Commit

483dce0

•

1 Parent(s): 2e7920c

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -29,7 +29,7 @@ class VidToMePipeline(DiffusionPipeline):
         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,
@@ -87,7 +87,7 @@ class VidToMePipeline(DiffusionPipeline):
                 'global_merge_ratio': global_merge_ratio
             },
             'seed': seed,
-            'device': self.device,
             'float_precision': self.float_precision
         }
         return config

         self.pipe = pipe
         self.scheduler = scheduler
         self.model_key = model_key
+        super().__init__()
     def __call__(self, video_path=None, video_prompt=None, edit_prompt=None,
                  control_type="none", n_timesteps=50, guidance_scale=7.5,
                 'global_merge_ratio': global_merge_ratio
             },
             'seed': seed,
+            'device': "cuda",
             'float_precision': self.float_precision
         }
         return config