๐Ÿ”ฅInstall PyTorch (optional)

Install PyTorch (optional)

PyTorch is required to run the transformer scanner.

If you do not install PyTorch, remove the transformer scanner from input_scanners field of the configuration file.

Use the selector on Pytorch's website to enter your system details and get the proper download link or for a quick start using CPU only, you can run:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

Last updated