Update controlnet_module.py
Browse files- controlnet_module.py +2 -2
controlnet_module.py
CHANGED
|
@@ -911,7 +911,7 @@ class ControlNetProcessor:
|
|
| 911 |
#=========================
|
| 912 |
|
| 913 |
# Masken speichern in Array
|
| 914 |
-
all_masks_crop = [] #Weiterverarbeitung in Crop-Größe
|
| 915 |
|
| 916 |
|
| 917 |
for i in range(num_masks):
|
|
@@ -919,7 +919,7 @@ class ControlNetProcessor:
|
|
| 919 |
# 2. FÜR VERARBEITUNG: Auf CROP-GRÖSSE interpolieren
|
| 920 |
resized_mask_crop = F.interpolate(
|
| 921 |
single_mask,
|
| 922 |
-
size=(image.height, image.width), # CROP-Größe bei Crop-Eingang sonst 512er-Größe
|
| 923 |
mode='bilinear',
|
| 924 |
align_corners=False
|
| 925 |
).squeeze()
|
|
|
|
| 911 |
#=========================
|
| 912 |
|
| 913 |
# Masken speichern in Array
|
| 914 |
+
all_masks_crop = [] #Weiterverarbeitung in Crop-Größe oder Bildgröße <= 512
|
| 915 |
|
| 916 |
|
| 917 |
for i in range(num_masks):
|
|
|
|
| 919 |
# 2. FÜR VERARBEITUNG: Auf CROP-GRÖSSE interpolieren
|
| 920 |
resized_mask_crop = F.interpolate(
|
| 921 |
single_mask,
|
| 922 |
+
size=(image.height, image.width), # CROP-Größe bei Crop-Eingang sonst <= 512er-Größe
|
| 923 |
mode='bilinear',
|
| 924 |
align_corners=False
|
| 925 |
).squeeze()
|