Wie konvertiert man das Safetensors-Modell in das ONNX-Modell?Python

Python-Programme
Anonymous
 Wie konvertiert man das Safetensors-Modell in das ONNX-Modell?

Post by Anonymous »

Ich möchte ein model.safetensors in ONNX konvertieren, leider habe ich nicht genügend Informationen zum Verfahren gefunden. Die Dokumentation des Pakets „safetensors“ reicht nicht aus und es ist tatsächlich nicht einmal klar, wie ich das Originalmodell (in meinem Fall „Pytorch“) erhalten kann, da ich so etwas wie
versuche

Code: Select all

with st.safe_open(modelsafetensors, framework="pt") as mystf:
...
Das mystf-Objekt hat get_tensor('sometensorname'), aber es scheint keine get_model()-Methode oder etwas Ähnliches zu haben.

Quick Reply

Change Text Case: 
   
  • Similar Topics
    Replies
    Views
    Last post