Closed Nomination-NRB closed 3 months ago
# line 248 pretrain = os.path.join(args.pretrained_model_name_or_path, 'unet') load_model_with_zeroed_mismatched_keys(unet, os.path.join(pretrain, 'diffusion_pytorch_model.safetensors'))
The diffusion_pytorch_model.safetensors file does not exist in the unet folder, only in the unet_encoder folder, do I need to change it to this:
pretrain = os.path.join(args.pretrained_model_name_or_path, 'unet_encoder') load_model_with_zeroed_mismatched_keys(unet, os.path.join(pretrain, 'diffusion_pytorch_model.safetensors'))
sorry, I fix it, pretrain = os.path.join(args.inpainting_model_path, 'unet')
# line 248 pretrain = os.path.join(args.pretrained_model_name_or_path, 'unet') load_model_with_zeroed_mismatched_keys(unet, os.path.join(pretrain, 'diffusion_pytorch_model.safetensors'))
The diffusion_pytorch_model.safetensors file does not exist in the unet folder, only in the unet_encoder folder, do I need to change it to this:
pretrain = os.path.join(args.pretrained_model_name_or_path, 'unet_encoder') load_model_with_zeroed_mismatched_keys(unet, os.path.join(pretrain, 'diffusion_pytorch_model.safetensors'))
sorry, I fix it, pretrain = os.path.join(args.inpainting_model_path, 'unet')
thanks
The diffusion_pytorch_model.safetensors file does not exist in the unet folder, only in the unet_encoder folder, do I need to change it to this: