codeysun commited on
Commit
ded11f1
·
verified ·
1 Parent(s): a64f64f

Upload pipeline.py

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