Skip to content

Commit 582755c

Browse files
committed
Merge remote-tracking branch 'origin/rf-inversion' into rf-inversion
2 parents 9d8b37c + 142aef5 commit 582755c

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

examples/community/pipeline_flux_rf_inversion.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@
5555
5656
>>> pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
5757
>>> pipe.to("cuda")
58-
58+
5959
>>> def download_image(url):
6060
... response = requests.get(url)
6161
... return PIL.Image.open(BytesIO(response.content)).convert("RGB")
@@ -777,10 +777,8 @@ def __call__(
777777

778778
# 6. Denoising loop
779779
with self.progress_bar(total=num_inference_steps) as progress_bar:
780-
781780
y_0 = image_latents.clone()
782781
for i, t in enumerate(timesteps):
783-
784782
t_i = 1 - t / 1000
785783
dt = torch.tensor(1 / (len(timesteps) - 1), device=device)
786784

0 commit comments

Comments
 (0)