codeysun commited on
Commit
b5f0fea
·
verified ·
1 Parent(s): 125d9fa

Upload pipeline.py

Browse files
Files changed (1) hide show
  1. 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