Update controlnet_module.py
Browse files- 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"""
|