Closed j217 closed 1 year ago
Fixed the issue with pip install --break-system-packages -q torch==2.1.1 Please make correction to your example code!
Looks like some distros have some outdated packages as the final version.
Since we went for a minimal setup (i.e. just torch.hub.load
instead of heavier packaging).
In any case, the docs explicitly say which PyTorch to use:
🐛 Bug
V4 TTS fails on vanilla Debian 12.
To Reproduce
Steps to reproduce the behavior:
$ python3 tts-v4 100.0% Traceback (most recent call last): File "/home/j217/tts-v4", line 13, in
model = torch.package.PackageImporter(local_file).load_pickle("tts_models", "model")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/lib/python3/dist-packages/torch/package/package_importer.py", line 271, in load_pickle
result = unpickler.load()
^^^^^^^^^^^^^^^^
File "/usr/lib/python3.11/pickle.py", line 1213, in load
dispatchkey[0]
File "/usr/lib/python3.11/pickle.py", line 1254, in load_binpersid
self.append(self.persistent_load(pid))
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/lib/python3/dist-packages/torch/package/package_importer.py", line 249, in persistent_load
loaded_reduces[reduce_id] = func(self, *args)
^^^^^^^^^^^^^^^^^
File "/usr/lib/python3/dist-packages/torch/jit/_script.py", line 372, in unpackage_script_module
cpp_module = torch._C._import_ir_module_from_package(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError:
Unknown builtin op: aten::scaled_dot_product_attention.
Here are some suggestions:
aten::_scaled_dot_product_attention
The original call is: File ".data/ts_code/code/torch/torch/nn/functional.py", line 489 _114 = [bsz, num_heads, src_len0, head_dim] v8 = torch.view(v6, _114) attn_output5 = torch.scaled_dot_product_attention(q3, k8, v8, attn_mask16, dropout_p0, is_causal)