lop / src /utils /safetensor_helper.py
thepianist9's picture
Upload folder using huggingface_hub
5a256aa verified
raw
history blame contribute delete
205 Bytes
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