silencer107
commited on
Commit
•
0ecc295
1
Parent(s):
2a4677b
Update src/pipeline.py
Browse files- 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.
|
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.
|
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]
|