NotImplementedError: prior not implemented for encoder of type

[protrolium]

I have trained with RAVE v2 and successfully exported a model without streaming, for the purposes of producing a prior model to use with nn~.

After attempting these hot fixes thanks to @fokklz, I am getting this error.

File "/content/download/src/acids-rave/scripts/train_prior.py", line 108, in main raise NotImplementedError("prior not implemented for encoder of type %s"%(type(pretrained.encoder))) NotImplementedError: prior not implemented for encoder of type <class 'rave.blocks.WasserteinEncoder'>

If anyone can shed some light on what this means or know of a solution I would greatly appreciate it.