What should I input for the network path in sample generation

[kujhin]

Hi, I am very interested in this task.

In the sample generation,

torchrun --standalone --nproc_per_node=1 generate.py --sampler_stages=both --outdir=./out_test
--network_first=/path/to/1st/ckpt --network_second=/path/to/2nd/ckpt

what should i input for --network_first and --network_second?

[tengjiayan20]

The first stage model can be any diffusion, and the second stage is the model trained by relay diffusion. For example, see section "Performance Reproduction" below.