Date about codes release

Question

Date about codes release

Closed this issue 2 months ago · 8 comments

when will you release codes? I read the paper, and it's a exciting work!

Answer 1 · 2024-01-25T01:51:27.000Z

Same question !

Answer 2 · 2024-02-04T19:32:36.000Z

Hello @zhouyuangan & @ziyizhetutanota, I've tried to produce a minimal working example here https://github.com/benearnthof/fm_boosting/tree/main Please feel free to check out the script there. I'm not too sure about the concatenation process that's depicted in Figure 3 of the paper so please raise issues or pull requests if you have suggestions. I used OT-CFM flow matching like it is implemented in TorchCFM https://github.com/atong01/conditional-flow-matching/tree/main

As of today this is just a minimal working example, training is slow but seems to work (I'll link samples tomorrow after 24hours of training on a A100 40GB GPU). I'll keep working on this and will wrap everything in a nice package in the coming days.

Answer 3 · 2024-03-05T16:23:03.000Z

Hi. How was the training? Is there any update?

Answer 4 · 2024-03-08T22:09:28.000Z

Hi @ygean, sorry for the delay. I've added noise concatenation with an image conditional UNet from imagen_pytorch by Phil Wang (Lucidrains on Github) and just started a training run on 128 -> 256 pixel images. Will take about two days. If you have any suggestions please feel free to raise an issue on my proof of concept implementation, so far results were blurry I'm hoping to get sharper images with the new UNet.

Answer 5 · 2024-03-11T14:27:12.000Z

Hi @ygean, sorry for the delay. I've added noise concatenation with an image conditional UNet from imagen_pytorch by Phil Wang (Lucidrains on Github) and just started a training run on 128 -> 256 pixel images. Will take about two days. If you have any suggestions please feel free to raise an issue on my proof of concept implementation, so far results were blurry I'm hoping to get sharper images with the new UNet.

Hi，does the new unet work fine now?

Answer 6 · 2024-03-12T18:32:45.000Z

Unfortunately not, my results are still blurry. I'll keep working on the implementation however as I want to integrate this model in a larger project of mine.

Answer 7 · 2024-07-17T12:51:15.000Z

when will you release codes?

Answer 8 · 2024-10-02T18:52:42.000Z

Hi we released the training code. In the next few days we will release the checkpoint and inference code.