File size: 282 Bytes
02984a5 |
1 2 3 4 5 6 |
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') |