Allow non tensor checkpoint values (#8845)

jackzhxng · web-flow · commit 160421a7c7a6 · 2025-03-03T11:46:46.000-08:00
diff --git a/examples/models/checkpoint.py b/examples/models/checkpoint.py
@@ -64,7 +64,7 @@ def get_checkpoint_dtype(checkpoint: Dict[str, Any]) -> Optional[str]:
         mismatched_dtypes = [
             (key, value.dtype)
             for key, value in checkpoint.items()
-            if value.dtype != dtype
+            if hasattr(value, "dtype") and value.dtype != dtype
         ]
         if len(mismatched_dtypes) > 0:
             print(

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ def get_checkpoint_dtype(checkpoint: Dict[str, Any]) -> Optional[str]:`
`64`	`64`	`mismatched_dtypes = [`
`65`	`65`	`(key, value.dtype)`
`66`	`66`	`for key, value in checkpoint.items()`
`67`		`- if value.dtype != dtype`
	`67`	`+ if hasattr(value, "dtype") and value.dtype != dtype`
`68`	`68`	`]`
`69`	`69`	`if len(mismatched_dtypes) > 0:`
`70`	`70`	`print(`