Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "1de2d0c8207b0f3b7f6119eeae5562fe57efeea9"
Unverified Commit 2a111bc9 authored by apolinário's avatar apolinário Committed by GitHub
Browse files

[Advanced Training Script] Fix pipe example (#6106)

parent 16e6997f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
View file @ 2a111bc9
...@@ -135,8 +135,8 @@ from safetensors.torch import load_file ...@@ -135,8 +135,8 @@ from safetensors.torch import load_file
""" """
diffusers_example_pivotal = f"""embedding_path = hf_hub_download(repo_id='{repo_id}', filename="embeddings.safetensors", repo_type="model") diffusers_example_pivotal = f"""embedding_path = hf_hub_download(repo_id='{repo_id}', filename="embeddings.safetensors", repo_type="model")
state_dict = load_file(embedding_path) state_dict = load_file(embedding_path)
pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer) pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2) pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
""" """
if token_abstraction_dict: if token_abstraction_dict:
for key, value in token_abstraction_dict.items(): for key, value in token_abstraction_dict.items():
......
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