| #!/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') |