linoyts HF staff commited on
Commit
6667957
1 Parent(s): 6e509af

Update clip_slider_pipeline.py

Browse files
Files changed (1) hide show
  1. clip_slider_pipeline.py +2 -2
clip_slider_pipeline.py CHANGED
@@ -17,8 +17,8 @@ class CLIPSlider:
17
  iterations: int = 300,
18
  ):
19
 
20
- self.device = device
21
- #self.pipe = sd_pipe.to(self.device)
22
  self.iterations = iterations
23
  if target_word != "" or opposite != "":
24
  self.avg_diff = self.find_latent_direction(target_word, opposite)
 
17
  iterations: int = 300,
18
  ):
19
 
20
+ #self.device = device
21
+ self.pipe = sd_pipe
22
  self.iterations = iterations
23
  if target_word != "" or opposite != "":
24
  self.avg_diff = self.find_latent_direction(target_word, opposite)