Astridkraft commited on
Commit
abf8b12
·
verified ·
1 Parent(s): cfd813d

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. 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=self.torch_dtype).to(self.device)
 
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