Astridkraft commited on
Commit
48aa073
·
verified ·
1 Parent(s): 153e917

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +1 -1
controlnet_module.py CHANGED
@@ -94,7 +94,7 @@ class ControlNetProcessor:
94
  return int(x1), int(y1), int(x2), int(y2)
95
 
96
 
97
- def create_sam_mask(self, image, bbox_coords, mode):
98
  """
99
  ERWEITERTE Funktion: Erstellt präzise Maske mit SAM 2
100
  """
 
94
  return int(x1), int(y1), int(x2), int(y2)
95
 
96
 
97
+ def create_sam_mask(self, image, bbox_coords, mode, is_front_face=False, is_back_head=False):
98
  """
99
  ERWEITERTE Funktion: Erstellt präzise Maske mit SAM 2
100
  """