Update controlnet_module.py

controlnet_module.py

@@ -50,11 +50,14 @@ class ControlNetProcessor:
             # Die korrekte Modell-ID für SAM 2 Tiny
             model_id = "facebook/sam2-hiera-tiny"
 
+
+                        
             # Lade Processor und Modell mit der neuen API
             self.sam_processor = Sam2Processor.from_pretrained(model_id)
-            self.sam_model = Sam2Model.from_pretrained(model_id).to(self.device)
+            self.sam_model = Sam2Model.from_pretrained(model_id, torch_dtype=self.torch_dtype).to(self.device)
             self.sam_model.eval()  # Setze Modell in Evaluierungsmodus
 
+            
             self.sam_initialized = True
             print("✅ SAM 2 erfolgreich geladen (via Transformers)")
             return True