Skip to content

Commit 14d2d79

Browse files
committed
fix: handling cases where pytorch file is a single tensor
1 parent c40b642 commit 14d2d79

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/core/handlers/pytorch/inspect.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,10 @@ def main():
3232
all_metadata = getattr(model, "_metadata") if hasattr(model, "_metadata") else {}
3333
model_metadata = all_metadata[""] if "" in all_metadata else {}
3434

35+
# some files are just a single tensor
36+
if isinstance(model, torch.Tensor):
37+
model = {"<unamed-single-tensor>": model}
38+
3539
inspection = {
3640
"file_path": file_path,
3741
"file_type": "PyTorch",

0 commit comments

Comments
 (0)