Upload addvpredkey.py
Browse files- addvpredkey.py +6 -0
addvpredkey.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from safetensors.torch import load_file, save_file
|
| 2 |
+
import torch
|
| 3 |
+
state_dict = load_file('./EasyFluffXL-Vnokey.safetensors')
|
| 4 |
+
state_dict['v_pred'] = torch.tensor([])
|
| 5 |
+
state_dict['ztsnr'] = torch.tensor([]) # Added for PR #16569
|
| 6 |
+
save_file(state_dict, './EasyFluffXL-V.safetensors')
|