diff --git a/yolo_world/models/detectors/yolo_world.py b/yolo_world/models/detectors/yolo_world.py index c72963d1..6c3767cf 100644 --- a/yolo_world/models/detectors/yolo_world.py +++ b/yolo_world/models/detectors/yolo_world.py @@ -58,7 +58,7 @@ def predict(self, def reparameterize(self, texts: List[List[str]]) -> None: # encode text embeddings into the detector self.texts = texts - self.text_feats, None = self.backbone.forward_text(texts) + self.text_feats, _ = self.backbone.forward_text(texts) def _forward( self,