Astridkraft commited on
Commit
ffb4be7
·
verified ·
1 Parent(s): 4ec52c9

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +24 -26
controlnet_module.py CHANGED
@@ -253,34 +253,32 @@ class ControlNetProcessor:
253
  print(f"Fehler beim Laden von Multi-ControlNet Outside: {e}")
254
  raise
255
  return self.pipe_multi_outside
256
-
257
-
258
- def prepare_controlnet_maps(self, image, keep_environment=False):
259
- """
260
- ERSTELLT NUR CONDITIONING-MAPS, generiert KEIN Bild.
261
- """
262
- print("🎯 ControlNet: Erstelle Conditioning-Maps...")
263
-
264
- if keep_environment:
265
- # Depth + Canny
266
- print(" Modus: Depth + Canny")
267
- conditioning_images = [
268
- self.extract_depth_map(image),
269
- self.extract_canny_edges(image)
270
- ]
271
- else:
272
- # OpenPose + Canny
273
- print(" Modus: OpenPose + Canny")
274
- conditioning_images = [
275
- self.extract_pose(image),
276
- self.extract_canny_edges(image)
277
- ]
278
-
279
- print(f"✅ {len(conditioning_images)} Conditioning-Maps erstellt.")
280
- return conditioning_images # Rückgabe: Liste der PIL Images
281
 
282
 
283
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
284
 
285
  def prepare_inpaint_input(self, image, keep_environment=False):
286
  """
 
253
  print(f"Fehler beim Laden von Multi-ControlNet Outside: {e}")
254
  raise
255
  return self.pipe_multi_outside
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
256
 
257
 
258
+ def prepare_controlnet_maps(self, image, keep_environment=False):
259
+ """
260
+ ERSTELLT NUR CONDITIONING-MAPS, generiert KEIN Bild.
261
+ """
262
+ print("🎯 ControlNet: Erstelle Conditioning-Maps...")
263
+
264
+ if keep_environment:
265
+ # Depth + Canny
266
+ print(" Modus: Depth + Canny")
267
+ conditioning_images = [
268
+ self.extract_depth_map(image),
269
+ self.extract_canny_edges(image)
270
+ ]
271
+ else:
272
+ # OpenPose + Canny
273
+ print(" Modus: OpenPose + Canny")
274
+ conditioning_images = [
275
+ self.extract_pose(image),
276
+ self.extract_canny_edges(image)
277
+ ]
278
+
279
+ print(f"✅ {len(conditioning_images)} Conditioning-Maps erstellt.")
280
+ return conditioning_images # Rückgabe: Liste der PIL Images
281
+
282
 
283
  def prepare_inpaint_input(self, image, keep_environment=False):
284
  """