Wie konvertiere ich das SafeTensors -Modell in das ONNX -Modell? ⇐ Python

[Anonymous]

Ich möchte ein Modell konvertieren. Die Dokumentation von SafeTensors Paket ist nicht genug und ist nicht klar, wie man das Originalmodell (Pytorch) bekommt, da ich etwas wie
versuche

Code: Select all

with st.safe_open(modelsafetensors, framework="pt") as mystf:
...

Das MYSTF -Objekt hat get_tensor ('slowensorname') , aber es scheint keine get_model () Methode oder ähnliches.