latentnavigation-flux

Sleeping

linoyts HF staff commited on Aug 22

Commit

f0f7887

•

1 Parent(s): ac3025f

Update clip_slider_pipeline.py

Files changed (1) hide show

clip_slider_pipeline.py CHANGED Viewed

@@ -373,7 +373,7 @@ class T5SliderFlux(CLIPSlider):
         with torch.no_grad():
             positives = []
             negatives = []
-            for i in tqdm(range(self.iterations)):
                 medium = random.choice(MEDIUMS)
                 subject = random.choice(SUBJECTS)
                 pos_prompt = f"a {medium} of a {target_word} {subject}"

         with torch.no_grad():
             positives = []
             negatives = []
+            for i in tqdm(range(iterations)):
                 medium = random.choice(MEDIUMS)
                 subject = random.choice(SUBJECTS)
                 pos_prompt = f"a {medium} of a {target_word} {subject}"