Update controlnet_module.py
Browse files- controlnet_module.py +2 -1
controlnet_module.py
CHANGED
|
@@ -54,7 +54,8 @@ class ControlNetProcessor:
|
|
| 54 |
|
| 55 |
# Lade Processor und Modell mit der neuen API
|
| 56 |
self.sam_processor = Sam2Processor.from_pretrained(model_id)
|
| 57 |
-
self.sam_model = Sam2Model.from_pretrained(model_id, torch_dtype=
|
|
|
|
| 58 |
self.sam_model.eval() # Setze Modell in Evaluierungsmodus
|
| 59 |
|
| 60 |
|
|
|
|
| 54 |
|
| 55 |
# Lade Processor und Modell mit der neuen API
|
| 56 |
self.sam_processor = Sam2Processor.from_pretrained(model_id)
|
| 57 |
+
self.sam_model = Sam2Model.from_pretrained(model_id, torch_dtype=torch.float32).to(self.device)
|
| 58 |
+
#self.sam_model = Sam2Model.from_pretrained(model_id, torch_dtype=self.torch_dtype).to(self.device)
|
| 59 |
self.sam_model.eval() # Setze Modell in Evaluierungsmodus
|
| 60 |
|
| 61 |
|