SeeSR icon indicating copy to clipboard operation
SeeSR copied to clipboard

The checkpoints can not load

Open Explore008 opened this issue 1 year ago • 2 comments

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.

Explore008 avatar Dec 27 '23 05:12 Explore008