search

cswry / SeeSR

[CVPR2024] SeeSR: Towards Semantics-Aware Real-World Image Super-Resolution
Apache License 2.0
438 stars 29 forks source link

The checkpoints can not load #5

Open Explore008 opened 11 months ago

Explore008 commented 11 months ago

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: c9c8268b3ac5bc3ccdcd6344d6b011b e2581c091d9f8b5a7ef3c6f73acbc65 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.

cswry commented 11 months ago

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 commented 4 months ago

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: c9c8268b3ac5bc3ccdcd6344d6b011b e2581c091d9f8b5a7ef3c6f73acbc65 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?