linoyts HF staff commited on
Commit
53bf1d0
1 Parent(s): 666c6de

Update clip_slider_pipeline.py

Browse files
Files changed (1) hide show
  1. clip_slider_pipeline.py +4 -1
clip_slider_pipeline.py CHANGED
@@ -20,7 +20,10 @@ class CLIPSlider:
20
  self.device = device
21
  self.pipe = sd_pipe.to(self.device)
22
  self.iterations = iterations
23
- self.avg_diff = self.find_latent_direction(target_word, opposite)
 
 
 
24
  if target_word_2nd != "" or opposite_2nd != "":
25
  self.avg_diff_2nd = self.find_latent_direction(target_word_2nd, opposite_2nd)
26
  else:
 
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)
25
+ else:
26
+ self.avg_diff = None
27
  if target_word_2nd != "" or opposite_2nd != "":
28
  self.avg_diff_2nd = self.find_latent_direction(target_word_2nd, opposite_2nd)
29
  else: