Skip to content

Commit bd69a73

Browse files
authored
fix is metadata func (#3220)
1 parent 086150b commit bd69a73

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/prototype/safetensors/safetensors_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def is_metadata_torchao(metadata: Dict[str, Any]):
209209

210210
# returns None if _type not in tensor_dict
211211
tensor_type = tensor_dict.get("_type")
212-
if tensor_type not in ALLOWED_TENSORS_SUBCLASSES or tensor_type != "Tensor":
212+
if tensor_type not in ALLOWED_TENSORS_SUBCLASSES and tensor_type != "Tensor":
213213
return False
214214

215215
return True

0 commit comments

Comments
 (0)