Views
No views yet
1from safetensors.torch import load_file 2 3state_dict = load_file("model.safetensors") 4model.load_state_dict(state_dict) 5model.eval()