Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -442,7 +442,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
442 |
num_inference_steps: int = 4,
|
443 |
timesteps: List[int] = None,
|
444 |
guidance_scale: float = 3.5,
|
445 |
-
lora_scale: Optional[
|
446 |
num_images_per_prompt: Optional[int] = 1,
|
447 |
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
|
448 |
latents: Optional[torch.FloatTensor] = None,
|
|
|
442 |
num_inference_steps: int = 4,
|
443 |
timesteps: List[int] = None,
|
444 |
guidance_scale: float = 3.5,
|
445 |
+
lora_scale: Optional[torch.FloatTensor] = None,
|
446 |
num_images_per_prompt: Optional[int] = 1,
|
447 |
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
|
448 |
latents: Optional[torch.FloatTensor] = None,
|