[pytorch] support pytorch 2.1.2 for Neuron

deepjavalibrary / djl

An Engine-Agnostic Deep Learning Framework in Java

https://djl.ai

Apache License 2.0

4.16k stars 661 forks source link

Closed siddvenk closed 2 days ago

siddvenk commented 2 days ago

Add PT 2.1.2 since Neuron still depends on it.

Remove PT 2.4.0 from staging publish step.