Based on the error message it seems that pickle is failing to copy the object, if it has a class attribute that references a module 11. I don’t know, if this is the case for QAT or if this is caused by your custom model, but could you check for module references inside your model?