Astridkraft commited on
Commit
d38831a
·
verified ·
1 Parent(s): f05d456

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +2 -2
controlnet_module.py CHANGED
@@ -888,8 +888,8 @@ class ControlNetProcessor:
888
  inputs = self.sam_processor(
889
  image_np,
890
  input_boxes=input_boxes,
891
- input_points=input_points, # ZWEI Punkte
892
- input_labels=input_labels, # Zwei Labels
893
  return_tensors="pt"
894
  ).to(self.device) # Ohne .to(self.device) werden die Tensoren standardmäßig im CPU-RAM erzeugt und gespeichert! Da GPU-Fehler!
895
 
 
888
  inputs = self.sam_processor(
889
  image_np,
890
  input_boxes=input_boxes,
891
+ input_points=input_points, # Ein Punkt
892
+ input_labels=input_labels, # Ein Label
893
  return_tensors="pt"
894
  ).to(self.device) # Ohne .to(self.device) werden die Tensoren standardmäßig im CPU-RAM erzeugt und gespeichert! Da GPU-Fehler!
895