Skip to content

Commit 37b70a7

Browse files
committed
fix import
1 parent 6d98da3 commit 37b70a7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/community/pipeline_flux_rf_inversion.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
scale_lora_layers,
3737
unscale_lora_layers,
3838
)
39+
from diffusers.utils.torch_utils import randn_tensor
3940

4041

4142
if is_torch_xla_available():
@@ -642,6 +643,7 @@ def __call__(
642643
start_timestep: float = 0,
643644
stop_timestep: float = 0.25,
644645
num_inference_steps: int = 28,
646+
sigmas: Optional[List[float]] = None,
645647
timesteps: List[int] = None,
646648
guidance_scale: float = 3.5,
647649
num_images_per_prompt: Optional[int] = 1,

0 commit comments

Comments
 (0)