Modifying code to output embedding

bbaaii / DreamDiffusion

Implementation of “DreamDiffusion: Generating High-Quality Images from Brain EEG Signals”

MIT License

429 stars 49 forks source link

Modifying code to output embedding #18

Open devs-cs opened 7 months ago

devs-cs commented 7 months ago

Does anyone have the pretrains/generation/checkpoint.pth file? Also, does anyone have advice on converting EEG to the latent embedding without the stable diffusion part?

alinvdu commented 7 months ago

Here you can find checkpoint: https://github.com/bbaaii/DreamDiffusion/commit/cbaf928d50808f988b7ffdf78d0d2f554448ee70