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