silencer107 commited on
Commit
0ecc295
1 Parent(s): 2a4677b

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +2 -2
src/pipeline.py CHANGED
@@ -806,7 +806,7 @@ class StableDiffusionXLPipeline(
806
  return_dict: bool = True,
807
  cross_attention_kwargs: Optional[Dict[str, Any]] = None,
808
  guidance_rescale: float = 0.0,
809
- end_cfg: float = 0.5,
810
  original_size: Optional[Tuple[int, int]] = None,
811
  crops_coords_top_left: Tuple[int, int] = (0, 0),
812
  target_size: Optional[Tuple[int, int]] = None,
@@ -1276,6 +1276,6 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
1276
  width=request.width,
1277
  height=request.height,
1278
  generator=generator,
1279
- end_cfg=0.5,
1280
  num_inference_steps=20,
1281
  ).images[0]
 
806
  return_dict: bool = True,
807
  cross_attention_kwargs: Optional[Dict[str, Any]] = None,
808
  guidance_rescale: float = 0.0,
809
+ end_cfg: float = 0.475,
810
  original_size: Optional[Tuple[int, int]] = None,
811
  crops_coords_top_left: Tuple[int, int] = (0, 0),
812
  target_size: Optional[Tuple[int, int]] = None,
 
1276
  width=request.width,
1277
  height=request.height,
1278
  generator=generator,
1279
+ end_cfg=0.475,
1280
  num_inference_steps=20,
1281
  ).images[0]