The checkpoints can not load

[Explore008]

Hello! Your work is excellent! In the process of replicating the experiment, I loaded the pre-training file on google drive, but the loading failed, I want to know how to solve it, thank you!
Traceback (most recent call last):
File "test_seesr.py", line 265, in
main(args)
File "test_seesr.py", line 167, in main
pipeline = load_seesr_pipeline(args, accelerator, enable_xformers_memory_efficient_attention)
File "test_seesr.py", line 83, in load_seesr_pipeline
unet = UNet2DConditionModel.from_pretrained(args.seesr_model_path, subfolder="unet")
File "/root/miniconda3/lib/python3.8/site-packages/diffusers/models/modeling_utils.py", line 646, in from_pretrained
raise ValueError(
ValueError: Cannot load <class 'models.unet_2d_condition.UNet2DConditionModel'> from preset/models/seesr because the following keys are missing:


The seesr-model_path is ''Seesr-main/preset/models/seesr''
I don't know if it's because of the safetensor file format.The command line parameters are the same as those in readme.

[Rongyuan Wu]

Hello, please verify if the placement of the pre-trained models matches the path mentioned in the inference command.

You need to consider the location of models such as SD2.1, SeeSR, RAM, and DAPE.

[rainsoulsrx]

Hello! Your work is excellent! In the process of replicating the experiment, I loaded the pre-training file on google drive, but the loading failed, I want to know how to solve it, thank you! Traceback (most recent call last): File "test_seesr.py", line 265, in main(args) File "test_seesr.py", line 167, in main pipeline = load_seesr_pipeline(args, accelerator, enable_xformers_memory_efficient_attention) File "test_seesr.py", line 83, in load_seesr_pipeline unet = UNet2DConditionModel.from_pretrained(args.seesr_model_path, subfolder="unet") File "/root/miniconda3/lib/python3.8/site-packages/diffusers/models/modeling_utils.py", line 646, in from_pretrained raise ValueError( ValueError: Cannot load <class 'models.unet_2d_condition.UNet2DConditionModel'> from preset/models/seesr because the following keys are missing: The seesr-model_path is ''Seesr-main/preset/models/seesr'' I don't know if it's because of the safetensor file format.The command line parameters are the same as those in readme.

Have you solved this?