codeysun commited on
Commit
9eb8207
·
verified ·
1 Parent(s): b5f0fea

Upload pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -1
pipeline.py CHANGED
@@ -1378,7 +1378,9 @@ class StableDiffusionControlNetPipeline(
1378
  if effective_region_mask is not None:
1379
  effective_region_mask = self.control_mask_processor.preprocess(
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(
 
1378
  if effective_region_mask is not None:
1379
  effective_region_mask = self.control_mask_processor.preprocess(
1380
  effective_region_mask, height=height, width=width
1381
+ ).to(dtype=prompt_embeds.dtype)
1382
+
1383
+ print(torch.unique(effective_region_mask))
1384
 
1385
  # 5. Prepare timesteps
1386
  timesteps, num_inference_steps = retrieve_timesteps(