Pin requirements to avoid error

openvinotoolkit / openvino_notebooks

📚 Jupyter notebook tutorials for OpenVINO™

Apache License 2.0

2.3k stars 791 forks source link

Pin requirements to avoid error #2196

Closed aleksandr-mokrov closed 2 months ago

aleksandr-mokrov commented 2 months ago

CVS-146676 Pin packages to fix error: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (tensor(1), tensor(1)) It happens not only during converting, but during an inference of the original model

review-notebook-app[bot] commented 2 months ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.