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

Upload pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -1
pipeline.py CHANGED
@@ -1378,15 +1378,17 @@ class StableDiffusionControlNetPipeline(
1378
  else:
1379
  assert False
1380
 
 
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
 
1378
  else:
1379
  assert False
1380
 
1381
+ # Prepare effective region mask
1382
  if effective_region_mask is not None:
1383
  effective_region_mask = self.control_mask_processor.preprocess(
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