Exporting .pth to coreml model

harrisonvanderbyl / rwkvstic

Framework agnostic python runtime for RWKV models

https://hazzzardous-rwkv-instruct.hf.space

MIT License

145 stars 18 forks source link

Exporting .pth to coreml model #11

Open abhishekmamdapure opened 1 year ago

abhishekmamdapure commented 1 year ago

When trying to convert RWKV("RWKV-4-Pile-169M-20220807-8023.pth")

receiving this error

   3178 index = inputs[2].val
   3180 assert dim.shape == ()
-> 3181 assert index.shape == ()
   3183 # NOTE:
   3184 # Each index in @begin_array/@end_array corresponds to a dimension of @_input
   3185 # Each val of those arrays corresponds to the start/end index to slice in that dimension
   3186 rank = _input.rank

AttributeError: 'NoneType' object has no attribute 'shape'

harrisonvanderbyl commented 1 year ago

the coreml option is currently a stub, I dont have an apple machine to test it on, so it is not fully implemented. If you or anybody is able to get it working, please submit a pull request