|
from latent_sdxl import get_solver as get_solver_sdxl |
|
set_seed(args.seed) |
|
create_workdir(args.workdir) |
|
|
|
solver_config = munchify({'num_sampling': args.NFE}) |
|
callback = None |
|
img = load_img(args.img_path, size=args.img_size) |
|
|
|
solver = get_solver(args.method, |
|
solver_config=solver_config, |
|
device=args.device) |
|
result = solver.sample(prompt=[args.null_prompt, args.prompt], |
|
src_img=img, |
|
cfg_guidance=args.cfg_guidance, |
|
callback_fn=callback) |
|
|
|
|
|
save_image(result, args.workdir.joinpath(f'result/reconstruct.png'), normalize=True) |
The --model argument is set to sdxl, but get_solver_sdxl is not being used, so SD1.5 is executed instead of SDXL.
