IDiff-Face icon indicating copy to clipboard operation
IDiff-Face copied to clipboard

Missing files when sampling with pre-trained model

Open Suimingzhe opened this issue 2 years ago • 5 comments

Thanks for your job! Could you please provide the file './hydra' and 'ema_averaged_model.ckpt' in pre-trained CPD25 and CPD50 ?

Suimingzhe avatar Aug 29 '23 11:08 Suimingzhe

After so many hours of debugging...I finally run the code successfully. The './hydra' and 'ema_averaged_model.ckpt' are unnecessary. Instead, I modify several lines of code in 'sample. py'. If you want the code, you can add my WeChat:130008841490. I will provide a detailed explanation for you.

NeRF-Factory avatar Aug 30 '23 10:08 NeRF-Factory

@NeRF-Factory Can you please explain, how did u change the sampling code

aksy2512 avatar Nov 03 '23 15:11 aksy2512

@aksy2512 Exactly, the checkpoint provided by the official repository is the weight of the 'eps_model' which is defined in the class 'DenoisingDiffusionProbabilisticModel'. You should write a new code for loading the checkpoint in the "sample.py". Additionally, the sampling process is also defined in 'DenoisingDiffusionProbabilisticModel'. You can find the code in "./diffusion/ddpm.py". This requires some modifications, but it's not too difficult.

I hope this help to answer your concern.

NeRF-Factory avatar Nov 28 '23 08:11 NeRF-Factory

@NeRF-Factory Could you please provide us with the code?

robbbgm avatar Jan 14 '24 00:01 robbbgm

The instructions for sampling have been updated. Check the requirements list and make sure to download all required models and configuration

fdbtrs avatar Feb 15 '24 11:02 fdbtrs