Spaces:
Sleeping
Sleeping
Zhen Ye
commited on
Commit
·
04d4562
1
Parent(s):
ed6bae2
Handle set-typed missing_keys in Grounding DINO loader
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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(
|