File size: 276 Bytes
04faaa5 |
1 2 3 4 5 6 7 8 9 |
#!/usr/bin/env python
import torch
from safetensors.torch import save_file
# pip install -r requirements.txt
model = torch.load('bad-hands-5.pt', map_location=torch.device('cpu'))
embedding_data = model['string_to_param']
save_file(embedding_data, 'bad-hands-5.safetensors') |