Upload pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -1381,13 +1381,13 @@ class StableDiffusionControlNetPipeline(
|
|
1381 |
if effective_region_mask is not None:
|
1382 |
effective_region_mask = self.control_mask_processor.preprocess(
|
1383 |
effective_region_mask, height=height, width=width
|
1384 |
-
)
|
1385 |
|
1386 |
print("mask shape:")
|
1387 |
print(effective_region_mask.shape)
|
1388 |
print()
|
1389 |
|
1390 |
-
print(max(effective_region_mask))
|
1391 |
|
1392 |
# 5. Prepare timesteps
|
1393 |
timesteps, num_inference_steps = retrieve_timesteps(
|
|
|
1381 |
if effective_region_mask is not None:
|
1382 |
effective_region_mask = self.control_mask_processor.preprocess(
|
1383 |
effective_region_mask, height=height, width=width
|
1384 |
+
)
|
1385 |
|
1386 |
print("mask shape:")
|
1387 |
print(effective_region_mask.shape)
|
1388 |
print()
|
1389 |
|
1390 |
+
print(torch.max(effective_region_mask))
|
1391 |
|
1392 |
# 5. Prepare timesteps
|
1393 |
timesteps, num_inference_steps = retrieve_timesteps(
|