AlekseyCalvin commited on
Commit
0369c77
·
verified ·
1 Parent(s): f1c229b

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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[Dict[str, Any]] = 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,
 
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,