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

Upload pipeline.py

Browse files
Files changed (1) hide show
  1. 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
- ).to(dtype=torch.float32)
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(