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') |