jadechoghari commited on
Commit
2ff21f6
1 Parent(s): 526c5d8

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +0 -1
pipeline.py CHANGED
@@ -22,7 +22,6 @@ class VidToMePipeline(DiffusionPipeline):
22
  self.register_to_config(device=device, sd_version=sd_version, float_precision=float_precision, height=height, width=width)
23
 
24
  # Now you can safely use self.device and other attributes
25
- self.device = torch.device(device if torch.cuda.is_available() else "cpu")
26
  self.sd_version = sd_version
27
  self.float_precision = float_precision
28
  self.height = height
 
22
  self.register_to_config(device=device, sd_version=sd_version, float_precision=float_precision, height=height, width=width)
23
 
24
  # Now you can safely use self.device and other attributes
 
25
  self.sd_version = sd_version
26
  self.float_precision = float_precision
27
  self.height = height