Astridkraft commited on
Commit
f3ad55c
·
verified ·
1 Parent(s): f4a8606

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +4 -1
controlnet_module.py CHANGED
@@ -25,7 +25,7 @@ class ControlNetProgressCallback:
25
  print(f"ControlNet Fortschritt: {self.current_step}/{self.total_steps} ({progress_percentage:.1%})")
26
  return callback_kwargs
27
 
28
-
29
  class ControlNetProcessor:
30
  def __init__(self, device="cuda", torch_dtype=torch.float32):
31
  self.device = device
@@ -39,6 +39,9 @@ class ControlNetProcessor:
39
  self.pipe_depth = None
40
  self.pipe_multi_inside = None # OpenPose + Canny für Inside-Box
41
  self.pipe_multi_outside = None # Depth + Canny für Outside-Box
 
 
 
42
 
43
  def load_pose_detector(self):
44
  """Lädt nur den Pose-Detector"""
 
25
  print(f"ControlNet Fortschritt: {self.current_step}/{self.total_steps} ({progress_percentage:.1%})")
26
  return callback_kwargs
27
 
28
+ #Initialisierung
29
  class ControlNetProcessor:
30
  def __init__(self, device="cuda", torch_dtype=torch.float32):
31
  self.device = device
 
39
  self.pipe_depth = None
40
  self.pipe_multi_inside = None # OpenPose + Canny für Inside-Box
41
  self.pipe_multi_outside = None # Depth + Canny für Outside-Box
42
+ self.conditioning_maps = None
43
+ self.controlnet_type = None
44
+ self.controlnet_scales = None
45
 
46
  def load_pose_detector(self):
47
  """Lädt nur den Pose-Detector"""