Spaces:
Runtime error
Runtime error
File size: 198 Bytes
e64d6ac |
1 2 3 4 5 6 7 8 |
def load_x_from_safetensor(checkpoint, key):
x_generator = {}
for k,v in checkpoint.items():
if key in k:
x_generator[k.replace(key+'.', '')] = v
return x_generator |