diff options
| author | PhytoEpidemic <64293310+PhytoEpidemic@users.noreply.github.com> | 2022-12-02 18:16:29 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-02 18:16:29 +0000 |
| commit | 119a945ef7569128eb7d6772468ffc5567c2e161 (patch) | |
| tree | 22080217fdece46fa461fdda215d7ac6c32c5871 /modules/textual_inversion/autocrop.py | |
| parent | 4b3c5bc24bffdf429c463a465763b3077fe55eb8 (diff) | |
| download | stable-diffusion-webui-gfx803-119a945ef7569128eb7d6772468ffc5567c2e161.tar.gz stable-diffusion-webui-gfx803-119a945ef7569128eb7d6772468ffc5567c2e161.tar.bz2 stable-diffusion-webui-gfx803-119a945ef7569128eb7d6772468ffc5567c2e161.zip | |
Fix divide by 0 error
Fix of the edge case 0 weight that occasionally will pop up in some specific situations. This was crashing the script.
Diffstat (limited to 'modules/textual_inversion/autocrop.py')
| -rw-r--r-- | modules/textual_inversion/autocrop.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/textual_inversion/autocrop.py b/modules/textual_inversion/autocrop.py index 9859974a..68e1103c 100644 --- a/modules/textual_inversion/autocrop.py +++ b/modules/textual_inversion/autocrop.py @@ -276,8 +276,8 @@ def poi_average(pois, settings): weight += poi.weight
x += poi.x * poi.weight
y += poi.y * poi.weight
- avg_x = round(x / weight)
- avg_y = round(y / weight)
+ avg_x = round(weight and x / weight)
+ avg_y = round(weight and y / weight)
return PointOfInterest(avg_x, avg_y)
@@ -338,4 +338,4 @@ class Settings: self.face_points_weight = face_points_weight
self.annotate_image = annotate_image
self.destop_view_image = False
- self.dnn_model_path = dnn_model_path
\ No newline at end of file + self.dnn_model_path = dnn_model_path
|