Add v_pred and ztsnr keys

#2
by aoleg - opened

The model is not detected as a v_pred model because the required keys are missing from the model's metadata; one needds to manually activate Advanced Sampling in order to use it. This can be easily avoided if the keys are added to metadata. The original script is published at https://civitai.com/models/1027203/noobvpencil-xl (check the "About this model" section).

Add v_pred and ztsnr keys.

# Python Script
BASE_MODEL_NAME = "noob_v_pencil-XL-v0.5.1"
tensors = {}

with safe_open(BASE_MODEL_NAME + "-base.safetensors", framework="pt", device="cpu") as f:
  for key in f.keys():
    tensors[key] = f.get_tensor(key)

tensors["v_pred"] = torch.tensor([0.0])
tensors["ztsnr"] = torch.tensor([0.0])
save_file(tensors, BASE_MODEL_NAME + ".safetensors")

thank you for your comment
this problem is fixed
I will reupload fixed model

Sign up or log in to comment