Skip to content

GitLab

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 ...@@ -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 ...@@ -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"
......
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