Zhen Ye commited on
Commit
04d4562
·
1 Parent(s): ed6bae2

Handle set-typed missing_keys in Grounding DINO loader

Browse files
Files changed (1) hide show
  1. models/detectors/grounding_dino.py +10 -1
models/detectors/grounding_dino.py CHANGED
@@ -97,7 +97,16 @@ class GroundingDinoDetector(ObjectDetector):
97
  # Backward compatibility with older transformers signatures.
98
  self.model = GroundingDinoForObjectDetection.from_pretrained(self.MODEL_NAME)
99
 
100
- missing_keys = load_info.get("missing_keys", []) if load_info else []
 
 
 
 
 
 
 
 
 
101
  if missing_keys:
102
  preview = ", ".join(missing_keys[:6])
103
  logging.warning(
 
97
  # Backward compatibility with older transformers signatures.
98
  self.model = GroundingDinoForObjectDetection.from_pretrained(self.MODEL_NAME)
99
 
100
+ missing_keys_raw = load_info.get("missing_keys", []) if load_info else []
101
+ if isinstance(missing_keys_raw, set):
102
+ # Some transformers builds return sets for loading_info fields.
103
+ missing_keys = sorted(str(key) for key in missing_keys_raw)
104
+ elif isinstance(missing_keys_raw, (list, tuple)):
105
+ missing_keys = [str(key) for key in missing_keys_raw]
106
+ elif missing_keys_raw:
107
+ missing_keys = [str(missing_keys_raw)]
108
+ else:
109
+ missing_keys = []
110
  if missing_keys:
111
  preview = ", ".join(missing_keys[:6])
112
  logging.warning(