upload rapunzel weights

experiments/single-concept/rapunzel/0064_rapunzel_ortho.yml

@@ -0,0 +1,98 @@
+# GENERATE TIME: Wed Jun 12 13:14:34 2024
+# CMD:
+# train_edlora.py -opt single-concept/train_configs/0064_rapunzel_ortho.yml
+
+name: 0064_rapunzel_ortho
+manual_seed: 64
+mixed_precision: fp16
+gradient_accumulation_steps: 1
+datasets:
+    train:
+        name: LoraDataset
+        concept_list: single-concept/data_configs/rapunzel.json
+        use_caption: true
+        use_mask: true
+        instance_transform:
+        -   type: HumanResizeCropFinalV3
+            size: 512
+            crop_p: 0.5
+        -   type: ToTensor
+        -   type: Normalize
+            mean:
+            - 0.5
+            std:
+            - 0.5
+        -   type: ShuffleCaption
+            keep_token_num: 1
+        -   type: EnhanceText
+            enhance_type: human
+        replace_mapping:
+            <TOK>: <rapunzel1> <rapunzel2>
+        batch_size_per_gpu: 2
+        dataset_enlarge_ratio: 500
+    val_vis:
+        name: PromptDataset
+        prompts: single-concept/validation_prompts/characters/test_girl_disney.txt
+        num_samples_per_prompt: 8
+        latent_size:
+        - 4
+        - 64
+        - 64
+        replace_mapping:
+            <TOK>: <rapunzel1> <rapunzel2>
+        batch_size_per_gpu: 4
+models:
+    pretrained_path: nitrosocke/mo-di-diffusion
+    enable_edlora: true
+    finetune_cfg:
+        text_embedding:
+            enable_tuning: true
+            lr: 0.001
+        text_encoder:
+            enable_tuning: true
+            lora_cfg:
+                rank: 5
+                alpha: 1.0
+                where: CLIPAttention
+            lr: 1.0e-05
+        unet:
+            enable_tuning: true
+            lora_cfg:
+                rank: 5
+                alpha: 1.0
+                where: Attention
+            lr: 0.0001
+    new_concept_token: <rapunzel1>+<rapunzel2>
+    initializer_token: <rand-0.013>+girl
+    noise_offset: 0.01
+    attn_reg_weight: 0.01
+    reg_full_identity: false
+    use_mask_loss: true
+    gradient_checkpoint: false
+    enable_xformers: true
+path:
+    pretrain_network: null
+train:
+    optim_g:
+        type: AdamW
+        lr: 0.0
+        weight_decay: 0.01
+        betas:
+        - 0.9
+        - 0.999
+    unet_kv_drop_rate: 0
+    scheduler: linear
+    emb_norm_threshold: 0.55
+val:
+    val_during_save: false
+    compose_visualize: false
+    alpha_list:
+    - 0
+    - 0.7
+    - 1.0
+    sample:
+        num_inference_steps: 50
+        guidance_scale: 7.5
+logger:
+    print_freq: 10
+    save_checkpoint_freq: 10000.0

experiments/single-concept/rapunzel/models/edlora_model-latest.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65972412f6eb9fa22f641d460cc0839a4b03e5cdfefcf4983632ff875d231adc
+size 35173046