Upload pipeline.py
Browse files- pipeline.py +5 -1
pipeline.py
CHANGED
@@ -1017,6 +1017,10 @@ class StableDiffusionControlNetPipeline(
|
|
1017 |
def apply_effective_region_mask(
|
1018 |
self, effective_region_mask: torch.Tensor, out: torch.Tensor
|
1019 |
) -> torch.Tensor:
|
|
|
|
|
|
|
|
|
1020 |
if effective_region_mask is None:
|
1021 |
return out
|
1022 |
|
@@ -1383,7 +1387,7 @@ class StableDiffusionControlNetPipeline(
|
|
1383 |
print(effective_region_mask.shape)
|
1384 |
print()
|
1385 |
|
1386 |
-
print(effective_region_mask)
|
1387 |
|
1388 |
# 5. Prepare timesteps
|
1389 |
timesteps, num_inference_steps = retrieve_timesteps(
|
|
|
1017 |
def apply_effective_region_mask(
|
1018 |
self, effective_region_mask: torch.Tensor, out: torch.Tensor
|
1019 |
) -> torch.Tensor:
|
1020 |
+
print("downblock dtype")
|
1021 |
+
print(out.dtype)
|
1022 |
+
print("mask dtype")
|
1023 |
+
print(effective_region_mask.dtype)
|
1024 |
if effective_region_mask is None:
|
1025 |
return out
|
1026 |
|
|
|
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(
|