Views
No views yet
pvc to trigger the image generation.nendoroid to trigger the image generation.figma to trigger the image generation.1model:
2 checkpoint_path: "./models/animagine-xl-4.0-opt.bnb_nf4.safetensors"
3 pretrained_model_name_or_path: cagliostrolab/animagine-xl-4.0
4
5 dtype: bfloat16
6
7 denoiser:
8 attn_implementation: "flash_attention_2"
9
10peft:
11 config:
12 type: lora
13 rank: 2
14 alpha: 1.0
15 dropout: 0.0
16
17 dtype: bfloat16
18
19 include_keys:
20 - "attn1"
21 - "attn2"
22 - ".ff."
23 exclude_keys: ["text_encoder", "vae"]
24
25dataset:
26 folder: "data/pvc"
27 num_repeats: 4
28 batch_size: 4
29
30 bucket_base_size: 1024
31 step: 128
32 min_size: 384
33 do_upscale: true
34
35 caption_processors:
36 - type: shuffle
37 split_separator: ","
38
39optimizer:
40 name: "schedulefree.RAdamScheduleFree"
41 args:
42 lr: 0.005
43
44scheduler:
45 # name: "torch.optim.lr_scheduler.ConstantLR"
46 # args: {}
47
48tracker:
49 project_name: "sdxl-pvc-1"
50 loggers:
51 - wandb
52
53saving:
54 strategy:
55 per_epochs: 0.25
56 per_steps: null
57 save_last: true
58
59 callbacks:
60 - type: "safentensors"
61 name: "sdxl-pvc"
62 save_dir: "./output/sdxl-pvc-2"
63
64preview:
65 strategy:
66 per_epochs: 1
67 per_steps: 100
68
69 callbacks:
70 - type: "local"
71 save_dir: "./output/sdxl-lora/preview"
72
73 data:
74 path: "./projects/pvc-sdxl/preview.yml"
75
76seed: 42
77num_train_epochs: 10
78
79trainer:
80 # debug_mode: "1step"
81
82 gradient_checkpointing: true
83
84 torch_compile: true
85 torch_compile_args:
86 backend: eager
87 mode: default
88 fullgraph: false
89
90 fp32_matmul_precision: "high"
91 allow_tf32: true1- prompt: |-
2 pvc style, 1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings,
3 green background, hat, hoop earrings, jewelry, looking at viewer,
4 shirt, short hair, simple background, solo, upper body, yellow shirt,
5 masterpiece, high score, great score, absurdres
6 negative_prompt: |-
7 lowres, bad anatomy, bad hands, text, error, missing finger, extra digits,
8 fewer digits, cropped, worst quality, low quality, low score, bad score,
9 average score, signature, watermark, username, blurry
10 height: 1024
11 width: 1024
12 cfg_scale: 5.0
13 num_steps: 20
14
15 seed: 0