detoxify icon indicating copy to clipboard operation
detoxify copied to clipboard

Running this on mobile with pytorch mobile

Open lee-hodg opened this issue 2 years ago • 1 comments

Is there anyway I can load the weights and "quantize" to reduce the size so I can run with pytorch mobile client side?

lee-hodg avatar Aug 10 '22 18:08 lee-hodg

Hello,

Currently all our models are trained with full precision so reducing the precision only at inference time could be unsafe. However, it should be possible and the torch docs on quantization should be helpful. We are planning to look more into a mobile version of Detoxify in the future, and of course, contributions are welcome!

laurahanu avatar Sep 09 '22 10:09 laurahanu