tensorflow-onnx icon indicating copy to clipboard operation
tensorflow-onnx copied to clipboard

What does it actually do?

Open f1am3d opened this issue 1 year ago • 4 comments

What does it actually do?

Question

The description is unclear.

  1. What does this repository actually do?
  2. Can I use it to convert ckpt to onnx?
  3. Can I use it to convert safetensors to onnx?
  4. Can it be used for stable diffustion in generat?

I am asking because I used this official AMD manual to convert SD to ONNX, but I need to convert custom model to ONNX and I am searching for a way how to do it.

f1am3d avatar Aug 31 '23 08:08 f1am3d

For ckpt and safetensors, looks like you are looking for ways to convert wights (instead of model structures) to ONNX representation. You may consider https://github.com/justinchuby/onnx-safetensors for working with safetensors.

justinchuby avatar Sep 12 '23 16:09 justinchuby

@f1am3d Did you find a way to convert a safetensors file to onnx? I am trying to achieve the same as you, but even with the onnx-safetensors extension i am not able to convert a .safetensors file to onnx. Could you help me with that if you found a solution?

Joly0 avatar Oct 08 '23 23:10 Joly0

@Joly0 You can share the script you used here. If it is a bug in onnx-safetensors I am happy to fix that.

justinchuby avatar Oct 09 '23 01:10 justinchuby

What does it actually do?

Question

The description is unclear.

  1. What does this repository actually do?
  2. Can I use it to convert ckpt to onnx?
  3. Can I use it to convert safetensors to onnx?
  4. Can it be used for stable diffustion in generat?

I am asking because I used this official AMD manual to convert SD to ONNX, but I need to convert custom model to ONNX and I am searching for a way how to do it.

Yes, this tool aims to help TensorFlow users to convert their models to ONNX, no matter public models or custom models.

Then answers to question 2) and 4) are Yes, as long as they are developed by TensorFlow.

fatcat-z avatar Oct 11 '23 06:10 fatcat-z