Spaces:

thepianist9
/

lop

Runtime error

lop / src /utils /safetensor_helper.py

Upload folder using huggingface_hub

5a256aa verified 3 months ago

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