We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c40b642 commit 14d2d79Copy full SHA for 14d2d79
src/core/handlers/pytorch/inspect.py
@@ -32,6 +32,10 @@ def main():
32
all_metadata = getattr(model, "_metadata") if hasattr(model, "_metadata") else {}
33
model_metadata = all_metadata[""] if "" in all_metadata else {}
34
35
+ # some files are just a single tensor
36
+ if isinstance(model, torch.Tensor):
37
+ model = {"<unamed-single-tensor>": model}
38
+
39
inspection = {
40
"file_path": file_path,
41
"file_type": "PyTorch",
0 commit comments