OSError: Error no file named diffusion_pytorch_model.bin found in directory preset/models/seesr.

[crazydiamondaz]

Thanks for your greatwork. I put the weight in the dir like this: /SeeSR-main/preset/models --DAPE.pth --seesr --stable-diffusion-2-base And run the command: python test_seesr.py \ --pretrained_model_path preset/models/stable-diffusion-2-base \ --prompt None \ --seesr_model_path preset/models/seesr \ --ram_ft_path preset/models/DAPE.pth \ --image_path preset/datasets/test_datasets \ --output_dir preset/datasets/output \ --start_point lr \ --num_inference_steps 50 \ --guidance_scale 5.5 \ --process_size 512 However, it turns out to be OSError:Error no file named diffusion_pytorch_model.bin found in directory preset/models/seesr. How can i fix this problem? Thanks a lot.

[huangyang-666]

Hello, I encountered the same problem as you. How did you solve it? Thank you!

[ningbende]

me too!

[dillfrescott]

Same issue!

[shams2023]

+1

[cswry]

Hello, can you provide details about this issue? @shams2023 @dillfrescott @ningbende @huangyang-666

[cswry]

Hello, @shams2023 @dillfrescott @ningbende @huangyang-666

I have updated the inference code.

Can you replace the code at line 83 unet = UNet2DConditionModel.from_pretrained(args.seesr_model_path, subfolder="unet") with code below in test_seesr.py? unet = UNet2DConditionModel.from_pretrained_orig(args.pretrained_model_path, args.seesr_model_path, subfolder="unet", use_image_cross_attention=True)

If it is not work for you, please tell me, thank you.

[huangyang-666]

Thank you for your prompt reply. I solved the issue by manually downloading each file of seesr and placing them in the corresponding location under 'preset/models/seesr'.