Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "026841447cd2e143b6d5d2fa9621b3bd1d975e25"
Commit 1598a579 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 63805f8a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
scripts/convert_lora_safetensor_to_diffusers.py
View file @ 1598a579
......@@ -24,7 +24,6 @@ from diffusers import StableDiffusionPipeline
def convert(base_model_path, checkpoint_path, LORA_PREFIX_UNET, LORA_PREFIX_TEXT_ENCODER, alpha):
# load base model
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
for key in state_dict:
# 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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment