Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make style

This commit is contained in:
Patrick von Platen 2023-03-06 10:51:03 +00:00
parent 63805f8af7
commit 1598a57958
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/convert_lora_safetensor_to_diffusers.py
View File

@ -24,7 +24,6 @@ from diffusers import StableDiffusionPipeline
def convert(base_model_path, checkpoint_path, LORA_PREFIX_UNET, LORA_PREFIX_TEXT_ENCODER, alpha): def convert(base_model_path, checkpoint_path, LORA_PREFIX_UNET, LORA_PREFIX_TEXT_ENCODER, alpha):
# load base model # load base model
pipeline = StableDiffusionPipeline.from_pretrained(base_model_path, torch_dtype=torch.float32) pipeline = StableDiffusionPipeline.from_pretrained(base_model_path, torch_dtype=torch.float32)
@ -35,7 +34,6 @@ def convert(base_model_path, checkpoint_path, LORA_PREFIX_UNET, LORA_PREFIX_TEXT
# directly update weight in diffusers model # directly update weight in diffusers model
for key in state_dict: for key in state_dict:
# it is suggested to print out the key, it usually will be something like below # it is suggested to print out the key, it usually will be something like below
# "lora_te_text_model_encoder_layers_0_self_attn_k_proj.lora_down.weight" # "lora_te_text_model_encoder_layers_0_self_attn_k_proj.lora_down.weight"