Views
No views yet
to trigger concept `<sksman>` → use `<s0><s1><s2><s3><s4><s5>` in your prompt
1from diffusers import AutoPipelineForText2Image
2import torch
3from huggingface_hub import hf_hub_download
4 from safetensors.torch import load_file
5
6pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
7pipeline.load_lora_weights('Tyl3rDrden/trained-flux-dev-dreambooth-JonathanTRAIN_TI_SMALL_PT_WITH_T5_V3', weight_name='pytorch_lora_weights.safetensors')
8embedding_path = hf_hub_download(repo_id='Tyl3rDrden/trained-flux-dev-dreambooth-JonathanTRAIN_TI_SMALL_PT_WITH_T5_V3', filename='trained-flux-dev-dreambooth-JonathanTRAIN_TI_SMALL_PT_WITH_T5_V3_emb.safetensors', repo_type="model")
9 state_dict = load_file(embedding_path)
10 pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>", "<s2>", "<s3>", "<s4>", "<s5>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
11 pipeline.load_textual_inversion(state_dict["t5"], token=["<s0>", "<s1>", "<s2>", "<s3>", "<s4>", "<s5>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
12
13image = pipeline('A photo of <s0><s1><s2><s3><s4><s5> in a bucket').images[0]# TODO: add an example code snippet for running this diffusion pipeline