EasyFluffXL / addvpredkey.py
zatochu's picture
Upload addvpredkey.py
02984a5 verified
raw
history blame contribute delete
282 Bytes
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')