有办法把Stable Diffusion的ckpt或者safetensors转成paddlenlp 能用的格式啊
这个不太会。要不去issue上问一下?
paddlenlp官方AI绘画项目那有ckpt或者safetensors转成ppdiffuser的实现,直接fork拿来用就得了
from safetensors.paddle import load_file import paddle import os def safetensors2pdpramas(filename): """ weights of safatensors to pdparams's """ device = 'gpu' if paddle.is_compiled_with_cuda() else 'cpu' if os.path.exists(filename.split('.safetensors')[0] + '.pdparams'): raise Exception('file existed!') w = load_file(filename, device=device) paddle.save(w, filename.split('.safetensors')[0] + '.pdparams') del w # 待转换文件路径 filenames = [ 'SDXL/model/text_encoder/model_state.safetensors', 'SDXL/model/text_encoder_2/model_state.safetensors', 'SDXL/model/unet/model_state.safetensors', 'SDXL/model/model_state.safetensors' ] # 开始转换 for filename in filenames: try: safetensors2pdparams(filename) except Exception as e: print(e) continue
这个不太会。要不去issue上问一下?
paddlenlp官方AI绘画项目那有ckpt或者safetensors转成ppdiffuser的实现,直接fork拿来用就得了