Astridkraft commited on
Commit
c1196be
·
verified ·
1 Parent(s): 7e1ad8c

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +2 -2
controlnet_module.py CHANGED
@@ -916,8 +916,8 @@ class ControlNetProcessor:
916
  inputs = self.sam_processor(
917
  image_np,
918
  input_boxes=input_boxes,
919
- input_points=input_points, # Ein Punkt
920
- input_labels=input_labels, # Ein Label
921
  return_tensors="pt"
922
  ).to(self.device) # Ohne .to(self.device) werden die Tensoren standardmäßig im CPU-RAM erzeugt und gespeichert! Da GPU-Fehler!
923
 
 
916
  inputs = self.sam_processor(
917
  image_np,
918
  input_boxes=input_boxes,
919
+ input_points=input_points, # Ein Punkt oder bei Hinterkopf 2
920
+ input_labels=input_labels, # Ein Label oder bei Hinterkopf 2
921
  return_tensors="pt"
922
  ).to(self.device) # Ohne .to(self.device) werden die Tensoren standardmäßig im CPU-RAM erzeugt und gespeichert! Da GPU-Fehler!
923