Fraser-Greenlee / transformer-vae

A library for making Transformer Variational Autoencoders. (Extends the Huggingface/transformers library.)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error on from_pretrained

pilot7747 opened this issue · comments

Hello!

I can't load the pretrained model due to the following error.
model = T5_VAE_Model.from_pretrained('t5-vae-poet')

OSError: Can't load config for 't5-vae-poet'. Make sure that:

- 't5-vae-poet' is a correct model identifier listed on 'https://huggingface.co/models'

- or 't5-vae-poet' is the correct path to a directory containing a config.json file

As I understand you don't provide the pre-trained model. Am I right?

Hey sorry for the late reply on this.

Yeah I don't have one ready yet but I will be training one soon!