Does touch.rb be able to running Core ML model ?

ankane / torch.rb

Deep learning for Ruby, powered by LibTorch

Other

704 stars 30 forks source link

Does touch.rb be able to running Core ML model ? #38

Closed Eric-Guo closed 1 year ago

Eric-Guo commented 1 year ago

https://github.com/apple/ml-stable-diffusion#-image-generation-with-python

I like ruby and plan to rebuild web ui in Rails as side project, so I want to know.

ankane commented 1 year ago

Hi @Eric-Guo, it's not currently possible to convert Torch.rb models to Core ML (this requires model tracing with torch.jit.trace, which isn't currently supported, but happy to review a PR if someone wants to add it).