Browse Source

Fix

pull/6227/head
AtomAlpaca 11 months ago
parent
commit
1d8fbf4ede
No known key found for this signature in database GPG Key ID: A003E4A074A5B2CA
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      tools/pnnx/src/load_torchscript.cpp

+ 1
- 2
tools/pnnx/src/load_torchscript.cpp View File

@@ -690,9 +690,8 @@ int load_torchscript(const std::string& ptpath, Graph& pnnx_graph,
input_tensors2.push_back(t);
}
}
else
else if(input_shapes2.size() != 0)
{
std::vector<at::Tensor> input_tensors2;
for (size_t i = 0; i < input_shapes2.size(); i++)
{
const std::vector<int64_t>& shape = input_shapes2[i];


Loading…
Cancel
Save