tsi-org commited on
Commit
f2c2f35
·
verified ·
1 Parent(s): b895b9f

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ image-0.png filter=lfs diff=lfs merge=lfs -text
37
+ image-1.png filter=lfs diff=lfs merge=lfs -text
38
+ image-2.png filter=lfs diff=lfs merge=lfs -text
39
+ image-3.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - stable-diffusion-xl
4
+ - stable-diffusion-xl-diffusers
5
+ - text-to-image
6
+ - diffusers
7
+ - lora
8
+ - template:sd-lora
9
+ widget:
10
+ - text: A <s0><s1> avatar
11
+ output:
12
+ url: image-0.png
13
+ - text: A <s0><s1> avatar
14
+ output:
15
+ url: image-1.png
16
+ - text: A <s0><s1> avatar
17
+ output:
18
+ url: image-2.png
19
+ - text: A <s0><s1> avatar
20
+ output:
21
+ url: image-3.png
22
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
23
+ instance_prompt: A <s0><s1> avatar
24
+ license: openrail++
25
+ ---
26
+
27
+ # SDXL LoRA DreamBooth - tsi-org/naavi-avatar
28
+
29
+ <Gallery />
30
+
31
+ ## Model description
32
+
33
+ ### These are tsi-org/naavi-avatar LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
34
+
35
+ ## Download model
36
+
37
+ ### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
38
+
39
+ - **LoRA**: download **[`naavi-avatar.safetensors` here 💾](/tsi-org/naavi-avatar/blob/main/naavi-avatar.safetensors)**.
40
+ - Place it on your `models/Lora` folder.
41
+ - On AUTOMATIC1111, load the LoRA by adding `<lora:naavi-avatar:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
42
+ - *Embeddings*: download **[`naavi-avatar_emb.safetensors` here 💾](/tsi-org/naavi-avatar/blob/main/naavi-avatar_emb.safetensors)**.
43
+ - Place it on it on your `embeddings` folder
44
+ - Use it by adding `naavi-avatar_emb` to your prompt. For example, `A naavi-avatar_emb avatar`
45
+ (you need both the LoRA and the embeddings as they were trained together for this LoRA)
46
+
47
+
48
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
49
+
50
+ ```py
51
+ from diffusers import AutoPipelineForText2Image
52
+ import torch
53
+ from huggingface_hub import hf_hub_download
54
+ from safetensors.torch import load_file
55
+
56
+ pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
57
+ pipeline.load_lora_weights('tsi-org/naavi-avatar', weight_name='pytorch_lora_weights.safetensors')
58
+ embedding_path = hf_hub_download(repo_id='tsi-org/naavi-avatar', filename='naavi-avatar_emb.safetensors' repo_type="model")
59
+ state_dict = load_file(embedding_path)
60
+ pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
61
+ pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
62
+
63
+ image = pipeline('A <s0><s1> avatar').images[0]
64
+ ```
65
+
66
+ For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)
67
+
68
+ ## Trigger words
69
+
70
+ To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
71
+
72
+ to trigger concept `TOK` → use `<s0><s1>` in your prompt
73
+
74
+
75
+
76
+ ## Details
77
+ All [Files & versions](/tsi-org/naavi-avatar/tree/main).
78
+
79
+ The weights were trained using [🧨 diffusers Advanced Dreambooth Training Script](https://github.com/huggingface/diffusers/blob/main/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py).
80
+
81
+ LoRA for the text encoder was enabled. False.
82
+
83
+ Pivotal tuning was enabled: True.
84
+
85
+ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
86
+
image-0.png ADDED

Git LFS Details

  • SHA256: be462cac9bc450e5df2ed48fdb44ce3e16174dbb432af2d09935546930343cb6
  • Pointer size: 132 Bytes
  • Size of remote file: 1.41 MB
image-1.png ADDED

Git LFS Details

  • SHA256: 68af001aa453b9451b5d6bae9d994418d643919655ff974aa58ab987962a2d0c
  • Pointer size: 132 Bytes
  • Size of remote file: 1.44 MB
image-2.png ADDED

Git LFS Details

  • SHA256: 236a85a1a302ea44b60141d9e32e6898d8736f644dd08c6fa30e97c9f7b42472
  • Pointer size: 132 Bytes
  • Size of remote file: 1.41 MB
image-3.png ADDED

Git LFS Details

  • SHA256: 217c548b4572dabe4fe02c82d36b6975a5ddf8594d6571f2032f476d663bcee1
  • Pointer size: 132 Bytes
  • Size of remote file: 1.43 MB
logs/dreambooth-lora-sd-xl/1710383473.1146913/events.out.tfevents.1710383473.r-tsi-org-autotrain-naavi-avatar-gfo0cdl4-ac1ba-xrcwt.177.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40f4f6e169ff7de58073d6b8b828e1c21a5f32f66586c810db3632421b840c81
3
+ size 3568
logs/dreambooth-lora-sd-xl/1710383473.1159716/hparams.yml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: null
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ cache_latents: true
9
+ caption_column: prompt
10
+ center_crop: false
11
+ checkpointing_steps: 100000
12
+ checkpoints_total_limit: null
13
+ class_data_dir: null
14
+ class_prompt: null
15
+ crops_coords_top_left_h: 0
16
+ crops_coords_top_left_w: 0
17
+ dataloader_num_workers: 0
18
+ dataset_config_name: null
19
+ dataset_name: ./6d5fd3df-b53c-4654-8e10-75dc098917c9
20
+ enable_xformers_memory_efficient_attention: false
21
+ gradient_accumulation_steps: 1
22
+ gradient_checkpointing: true
23
+ hub_model_id: null
24
+ hub_token: null
25
+ image_column: image
26
+ instance_data_dir: null
27
+ instance_prompt: A <s0><s1> avatar
28
+ learning_rate: 1.0
29
+ local_rank: -1
30
+ logging_dir: logs
31
+ lr_num_cycles: 1
32
+ lr_power: 1.0
33
+ lr_scheduler: constant
34
+ lr_warmup_steps: 0
35
+ max_grad_norm: 1.0
36
+ max_train_steps: 720
37
+ mixed_precision: bf16
38
+ noise_offset: 0
39
+ num_class_images: 100
40
+ num_new_tokens_per_abstraction: 2
41
+ num_train_epochs: 120
42
+ num_validation_images: 4
43
+ optimizer: prodigy
44
+ output_dir: naavi-avatar
45
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
46
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
47
+ prior_generation_precision: null
48
+ prior_loss_weight: 1.0
49
+ prodigy_beta3: null
50
+ prodigy_decouple: true
51
+ prodigy_safeguard_warmup: true
52
+ prodigy_use_bias_correction: true
53
+ push_to_hub: false
54
+ rank: 32
55
+ repeats: 3
56
+ report_to: tensorboard
57
+ resolution: 1024
58
+ resume_from_checkpoint: null
59
+ revision: null
60
+ sample_batch_size: 4
61
+ scale_lr: false
62
+ seed: 42
63
+ snr_gamma: null
64
+ text_encoder_lr: 1.0
65
+ token_abstraction: TOK
66
+ train_batch_size: 2
67
+ train_text_encoder: false
68
+ train_text_encoder_frac: 1.0
69
+ train_text_encoder_ti: true
70
+ train_text_encoder_ti_frac: 0.5
71
+ use_8bit_adam: false
72
+ validation_epochs: 50
73
+ validation_prompt: null
74
+ variant: null
75
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1710383473.r-tsi-org-autotrain-naavi-avatar-gfo0cdl4-ac1ba-xrcwt.177.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa231ce8f8a2569709e7949d15a8bd51c9036ed5a961ae2f2ab3dba7db636926
3
+ size 60314
naavi-avatar.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71428d3929c1322f58539a567757e1429ed6301183e3d373855e567dfb091ff3
3
+ size 186046568
naavi-avatar_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d09d1917612a7eb1e5317b6f434a17e0a5d9d0ea17cc7201071aba84259989b
3
+ size 16536
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:454b790881d1bb23e8acdc5602b19df61238a95afbda058aa067a86ca169d653
3
+ size 185963768