Fixed incorrect safetensors API call in README

README.md

@@ -49,12 +49,12 @@ import torch
 import torchvision.transforms as transforms
 from torchvision.utils import save_image
 from omegaconf import OmegaConf
-from safetensors.torch import load_model
+from safetensors.torch import load_file
 from olvae.utils import instantiate_from_config
 
 def load_model_from_config(config_path, ckpt_path, device=torch.device("cuda")):
 	config = OmegaConf.load(config_path)
-	sd = load_model(ckpt_path)
+	sd = load_file(ckpt_path)
 	model = instantiate_from_config(config.model)
 	model.load_state_dict(sd, strict=False)
 	model = model.to(device).eval()