from safetensors.torch import load_file, save_file import torch state_dict = load_file('./EasyFluffXL-Vnokey.safetensors') state_dict['v_pred'] = torch.tensor([]) state_dict['ztsnr'] = torch.tensor([]) # Added for PR #16569 save_file(state_dict, './EasyFluffXL-V.safetensors')