目前使用notebook版本的FaceChain
FaceChain训练的人物Lora:pytorch_lora_weights.bin似乎是一个二进制文件?
其他版本的FaceChain也是如此吗?
能否导出到本地部署的stable-diffusion中使用?
【 FaceChain训练的人物Lora能否导出safetensors文件使用?
】
是的,其他版本的 FaceChain 也是类似的。在训练过程中,人物 Lora 的权重通常会保存在一个二进制文件中,如您提到的 pytorch_lora_weights.bin。这个文件包含了模型在训练过程中学到的参数。
要将训练好的人物 Lora 导出到本地部署的 stable-diffusion 中使用,您需要执行以下步骤:
import torch
model = torch.load('pytorch_lora_model.pt')
weight = torch.load('pytorch_lora_weights.bin')
CopyCopy
from stable_baselines3 import PPO
from stable_baselines3.common.policies import ActorCriticPolicy
from stable_baselines3.common.env_util import make_vec_env
from stable_baselines3.common.callbacks import BaseCallback
from stable_baselines3.common.vec_env import VecNormalize
model = PPO(ActorCriticPolicy, env='Pendulum-v0', verbose=1)
model.load_model('pytorch_lora_model.pt', 'pytorch_lora_weights.bin')
env = make_vec_env('Pendulum-v0', n_envs=4)
class CustomCallback(BaseCallback):
def _on_step(self) -> bool:
return True
model.learn(env, total_timesteps=10000, callback=CustomCallback())
CopyCopy
这样,您就可以在本地部署的 stable-diffusion 中使用训练好的人物 Lora 了。
ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352