linoyts HF staff commited on
Commit
f0f7887
β€’
1 Parent(s): ac3025f

Update clip_slider_pipeline.py

Browse files
Files changed (1) hide show
  1. clip_slider_pipeline.py +1 -1
clip_slider_pipeline.py CHANGED
@@ -373,7 +373,7 @@ class T5SliderFlux(CLIPSlider):
373
  with torch.no_grad():
374
  positives = []
375
  negatives = []
376
- for i in tqdm(range(self.iterations)):
377
  medium = random.choice(MEDIUMS)
378
  subject = random.choice(SUBJECTS)
379
  pos_prompt = f"a {medium} of a {target_word} {subject}"
 
373
  with torch.no_grad():
374
  positives = []
375
  negatives = []
376
+ for i in tqdm(range(iterations)):
377
  medium = random.choice(MEDIUMS)
378
  subject = random.choice(SUBJECTS)
379
  pos_prompt = f"a {medium} of a {target_word} {subject}"