getao / icae

The repo for In-context Autoencoder
Creative Commons Zero v1.0 Universal
54 stars 2 forks source link

AutoEncoder Inference for Version 1 #14

Open Haoliu-cola opened 2 months ago

Haoliu-cola commented 2 months ago

Hi, thanks for your great work. I would like to test the autoencoding part in your v1 model. I tried to adjust line 89 in ft_inference.py to prompt_ids = [model.ae_token_id], but got empty generated outcomes. Could you advise on how to get a correct AE output? Thanks!