Upload pipeline.py
Browse files- pipeline.py +0 -11
pipeline.py
CHANGED
|
@@ -1017,10 +1017,6 @@ class StableDiffusionControlNetPipeline(
|
|
| 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 |
|
|
@@ -1384,13 +1380,6 @@ class StableDiffusionControlNetPipeline(
|
|
| 1384 |
effective_region_mask, height=height, width=width
|
| 1385 |
).to(dtype=torch.float16)
|
| 1386 |
|
| 1387 |
-
print("mask shape:")
|
| 1388 |
-
print(effective_region_mask.shape)
|
| 1389 |
-
print()
|
| 1390 |
-
|
| 1391 |
-
print(torch.min(effective_region_mask))
|
| 1392 |
-
print(torch.max(effective_region_mask))
|
| 1393 |
-
|
| 1394 |
# 5. Prepare timesteps
|
| 1395 |
timesteps, num_inference_steps = retrieve_timesteps(
|
| 1396 |
self.scheduler, num_inference_steps, device, timesteps, sigmas
|
|
|
|
| 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 |
|
|
|
|
| 1380 |
effective_region_mask, height=height, width=width
|
| 1381 |
).to(dtype=torch.float16)
|
| 1382 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1383 |
# 5. Prepare timesteps
|
| 1384 |
timesteps, num_inference_steps = retrieve_timesteps(
|
| 1385 |
self.scheduler, num_inference_steps, device, timesteps, sigmas
|