alc15492 commited on
Commit
cbc3de3
1 Parent(s): 1e97b03

feat: upload Reflections lora model

Browse files
Reflections_config/config_file.toml ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Realistic_Vision_V2.0.safetensors"
5
+ vae = "/content/vae/stablediffusion.vae.pt"
6
+
7
+ [additional_network_arguments]
8
+ no_metadata = false
9
+ unet_lr = 0.0005
10
+ network_module = "networks.lora"
11
+ network_dim = 128
12
+ network_alpha = 64
13
+ network_train_unet_only = true
14
+ network_train_text_encoder_only = false
15
+
16
+ [optimizer_arguments]
17
+ optimizer_type = "AdamW8bit"
18
+ learning_rate = 0.0005
19
+ max_grad_norm = 1.0
20
+ lr_scheduler = "constant"
21
+ lr_warmup_steps = 0
22
+
23
+ [dataset_arguments]
24
+ debug_dataset = false
25
+ in_json = "/content/LoRA/meta_lat.json"
26
+ train_data_dir = "/content/LoRA/train_data"
27
+ dataset_repeats = 1
28
+ shuffle_caption = true
29
+ keep_tokens = 0
30
+ resolution = "768,768"
31
+ caption_dropout_rate = 0
32
+ caption_tag_dropout_rate = 0
33
+ caption_dropout_every_n_epochs = 0
34
+ color_aug = false
35
+ token_warmup_min = 1
36
+ token_warmup_step = 0
37
+
38
+ [training_arguments]
39
+ output_dir = "/content/LoRA/output"
40
+ output_name = "Reflections"
41
+ save_precision = "fp16"
42
+ save_every_n_epochs = 25
43
+ train_batch_size = 6
44
+ max_token_length = 225
45
+ mem_eff_attn = false
46
+ xformers = true
47
+ max_train_epochs = 400
48
+ max_data_loader_n_workers = 8
49
+ persistent_data_loader_workers = true
50
+ gradient_checkpointing = false
51
+ gradient_accumulation_steps = 1
52
+ mixed_precision = "fp16"
53
+ clip_skip = 2
54
+ logging_dir = "/content/LoRA/logs"
55
+ log_prefix = "Reflections"
56
+ noise_offset = 0.3
57
+ lowram = true
58
+
59
+ [sample_prompt_arguments]
60
+ sample_every_n_epochs = 1
61
+ sample_sampler = "ddim"
62
+
63
+ [saving_arguments]
64
+ save_model_as = "safetensors"
Reflections_config/sample_prompt.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ blurry reflection of a building on water, bright lighting --w 768 --h 768 --l 7 --s 28