diff options
author | Nuullll <vfirst218@gmail.com> | 2023-12-06 20:55:42 +0800 |
---|---|---|
committer | Nuullll <vfirst218@gmail.com> | 2023-12-06 20:55:47 +0800 |
commit | 746783f7a47f38f728f221cc26fe04035d3ca66b (patch) | |
tree | 69a7b1c6aa2f2e20db0994ae58204b26694a296e /modules/prompt_parser.py | |
parent | f92d61497a426a19818625c3ccdaae9beeb82b31 (diff) |
[IPEX] Fix embedding
Cast `torch.bmm` args into same `dtype`.
Fixes the following error when using Text Inversion embedding (#14224):
```
RuntimeError: could not create a primitive descriptor for a matmul
primitive
```
Diffstat (limited to 'modules/prompt_parser.py')
0 files changed, 0 insertions, 0 deletions