Example code execution
Can you add an example to the README of how to work with your model?
I found this class in your GitHub repo: https://github.com/kandinskylab/kvae/blob/5b2dcdf05b6cdd1006e0f0378980601a8debe443/kvae/models/kvae_3d.py#L24 and probably this class could be integrated into this model to load with trust_remote_code.
Examples of working with the model via the official github repository have been added with complete instructions and ready-made scripts for running and calculating metrics.
By inheriting the KVAE model from the PyTorchModelHubMixin class, loading models directly from HuggingFace is supported (simply specify the model path to hf, as it is done in the GitHub launch script). Working via transformers.AutoModel is not implemented; this requires additional modifications to the model.